Update train.py

train.py

@@ -133,7 +133,9 @@ training_args = TrainingArguments(
     save_strategy="epoch",
     remove_unused_columns=False,
     gradient_checkpointing=True,
-    fp16=torch.cuda.is_available()
+    fp16=False,  # ✅ 禁用 fp16，避免 _scale=None 错误
+    bf16=True if torch.cuda.is_available() else False  # ✅ 仅在支持 bf16 时启用
+
 )
 
 # ✅ **手动禁用 `use_cache`**