Spaces:
Sleeping
Sleeping
Update finetune.py
Browse files- 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("
|
| 59 |
-
tokenizer.save_pretrained("
|
| 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.")
|