dangermouse77
/

FromAnswerToQuestion-T5-small-60M

Model card Files Files and versions

dangermouse77 commited on Mar 5, 2025

Commit

ff18de9

·

verified ·

1 Parent(s): a8fa556

Update finetune.py

Files changed (1) hide show

finetune.py +1 -1

finetune.py CHANGED Viewed

@@ -47,7 +47,7 @@ trainer = Trainer(
 trainer.train()
 # Save trained model
-output_dir = "/home/dm/chat/AQ/aq_model"  # Change the folder name if needed
 trainer.save_model(output_dir)
 tokenizer.save_pretrained(output_dir)

 trainer.train()
 # Save trained model
+output_dir = "./aq_model"  # Change the folder name if needed
 trainer.save_model(output_dir)
 tokenizer.save_pretrained(output_dir)