Răsfoiți Sursa

修复报错

lxylxy123321 1 zi în urmă
părinte
comite
029e5ccb97
1 a modificat fișierele cu 1 adăugiri și 0 ștergeri
  1. 1 0
      backend/app/engines/text_engine.py

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

@@ -416,6 +416,7 @@ class TextEngine(BaseEngine):
                                     },
                                     },
                                 )
                                 )
 
 
+            os.makedirs(output_dir, exist_ok=True)
             self._model.save_pretrained(output_dir)
             self._model.save_pretrained(output_dir)
             self._tokenizer.save_pretrained(output_dir)
             self._tokenizer.save_pretrained(output_dir)
             logger.info(f"PPO training completed for job {job_id}")
             logger.info(f"PPO training completed for job {job_id}")