完成了load1) er.py, 以 2) 部分代码的loader加载路径的更改

src/models/ocr_model/model/TexTeller.py

@@ -5,6 +5,7 @@ from models.globals import (
     VOCAB_SIZE,
     OCR_IMG_SIZE,
     OCR_IMG_CHANNELS,
+    MAX_TOKEN_SIZE
 )
 
 from transformers import (
@@ -25,6 +26,7 @@ class TexTeller(VisionEncoderDecoderModel):
         ))
         decoder = TrOCRForCausalLM(TrOCRConfig(
             vocab_size=VOCAB_SIZE,
+            max_position_embeddings=MAX_TOKEN_SIZE
         ))
         super().__init__(encoder=encoder, decoder=decoder)