Is the function in the code corresponding to the module in the picture like this? But after I finished writing, I found that the size of my model changed from 70M to about 7M in the original. However, the dLA34 in the paper is only 1.3M. Is there something wrong with my understanding?

Is the function in the code corresponding to the module in the picture like this? But after I finished writing, I found that the size of my model changed from 70M to about 7M in the original. However, the dLA34 in the paper is only 1.3M. Is there something wrong with my understanding?
