Jelajahi Sumber

修复报错

lxylxy123321 18 jam lalu
induk
melakukan
27b2c7c3d2
1 mengubah file dengan 1 tambahan dan 1 penghapusan
  1. 1 1
      backend/app/engines/text_engine.py

+ 1 - 1
backend/app/engines/text_engine.py

@@ -339,7 +339,7 @@ class TextEngine(BaseEngine):
                 ref_model=ref_model,
                 args=DPOConfig(**base_trainer_kwargs),
                 train_dataset=dataset,
-                tokenizer=self._tokenizer,
+                processing_class=self._tokenizer,
             )
 
             # 修复 Qwen tokenizer bug(TRL #1073):