lxylxy123321 преди 20 часа
родител
ревизия
5980af33a7
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      backend/app/engines/text_engine.py

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

@@ -333,7 +333,7 @@ class TextEngine(BaseEngine):
                 gradient_accumulation_steps=gradient_accumulation,
                 ppo_epochs=ppo_epochs,
                 vf_coef=vf_coef,
-                kl_ctl=kl_coef,
+                init_kl_coef=kl_coef,
                 response_length=response_length,
                 output_dir=output_dir,
                 logging_steps=10,