krisha06 commited on
Commit
cf7a2de
·
verified ·
1 Parent(s): f751957

Update finetune.py

Browse files
Files changed (1) hide show
  1. finetune.py +2 -2
finetune.py CHANGED
@@ -55,6 +55,6 @@ trainer = Trainer(
55
  trainer.train()
56
 
57
  # Save the fine-tuned model
58
- model.save_pretrained("./tinyllama_lora_finetuned")
59
- tokenizer.save_pretrained("./tinyllama_lora_finetuned")
60
  print("✅ Fine-tuning complete! Model saved.")
 
55
  trainer.train()
56
 
57
  # Save the fine-tuned model
58
+ model.save_pretrained("tinyllama-lora-finetuned")
59
+ tokenizer.save_pretrained("tinyllama-lora-finetuned")
60
  print("✅ Fine-tuning complete! Model saved.")