save llama trained models to correct location

transformerlab · Feb 21, 2024 · c8c1559 · c8c1559
1 parent 2255067
commit c8c1559
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/transformerlab/plugins/llama_trainer/main.py b/transformerlab/plugins/llama_trainer/main.py
@@ -157,4 +157,4 @@ def on_step_end(self, args, state, control, **kwargs):
 trainer.train()
 
 # save model
-trainer.save_model()
+trainer.save_model(output_dir=config["adaptor_output_dir"])