Parcourir la source

修复ppo报错

lxylxy123321 il y a 1 jour
Parent
commit
c31c29de96
1 fichiers modifiés avec 2 ajouts et 1 suppressions
  1. 2 1
      backend/app/engines/text_engine.py

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

@@ -323,7 +323,8 @@ class TextEngine(BaseEngine):
             self._model = AutoModelForCausalLMWithValueHead.from_pretrained(
                 self._model, peft_config=peft_config,
             )
-            self._model.print_trainable_parameters()
+            if hasattr(self._model, "print_trainable_parameters"):
+                self._model.print_trainable_parameters()
 
             # TRL 0.9.x PPOConfig 只接受 PPO 专用参数,不支持 HuggingFace Trainer 参数
             # mini_batch_size 必须满足:batch_size % (mini_batch_size * gradient_accumulation_steps) == 0