lxylxy123321 vor 21 Stunden
Ursprung
Commit
029e5ccb97
1 geänderte Dateien mit 1 neuen und 0 gelöschten Zeilen
  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._tokenizer.save_pretrained(output_dir)
             logger.info(f"PPO training completed for job {job_id}")