遵循面向对象思想重构train_model_grpo.py

src/model_downloader.py

@@ -11,7 +11,7 @@ from modelscope import snapshot_download
 # model_dir = snapshot_download('deepseek-ai/Janus-Pro-7B', cache_dir="../models/")
 
 # model_dir = snapshot_download('AI-ModelScope/bert-base-uncased', cache_dir="../models/")
-model_dir = snapshot_download('LLM-Research/longformer-base-4096', cache_dir="../models/")
+model_dir = snapshot_download('allenai/longformer-base-4096', cache_dir="../models/)
 
 # 验证SDK token
 # 据模型源上传人说，模型支持华为 昇腾(Ascend) 910

src/train_model_grpo_v1.1.py

@@ -27,8 +27,8 @@ class ModelTrainer:
         self.lora_rank = config.lora_rank
         self.gpu_memory_utilization = config.gpu_memory_utilization
         # 初始化 BERT 模型和分词器
-        self.tokenizer = BertTokenizer.from_pretrained(f'../models/LLM-Research/longformer-base-4096')
-        self.bert_model = BertModel.from_pretrained(f'../models/LLM-Research/longformer-base-4096')
+        self.tokenizer = BertTokenizer.from_pretrained(f'../models/allenai/longformer-base-4096')
+        self.bert_model = BertModel.from_pretrained(f'../models/allenai/longformer-base-4096')
 
     def load_model(self):
         """