The entire code has significant issues.

[yhz5613813]

I carefully went through the source code, and not only is there a bunch of errors, but even the model structure itself seems flawed. I suspect the author just haphazardly stitched together code from elsewhere on Hugging Face. As for the training part in train.py, I highly doubt it was properly coded either, because there are numerous inconsistencies in variable names throughout the code. Without modifications, it simply won't run.