Update train.py
Browse files
train.py
CHANGED
|
@@ -91,7 +91,7 @@ def train_model():
|
|
| 91 |
save_strategy="epoch", # Save model after each epoch
|
| 92 |
load_best_model_at_end=True, # Load best model at end of training
|
| 93 |
push_to_hub=True, # Push model to Hugging Face Hub
|
| 94 |
-
hub_model_id="
|
| 95 |
)
|
| 96 |
|
| 97 |
# Step 7: Initialize the trainer
|
|
|
|
| 91 |
save_strategy="epoch", # Save model after each epoch
|
| 92 |
load_best_model_at_end=True, # Load best model at end of training
|
| 93 |
push_to_hub=True, # Push model to Hugging Face Hub
|
| 94 |
+
hub_model_id="shaheerawan3/Vibescribe" # Replace with your username
|
| 95 |
)
|
| 96 |
|
| 97 |
# Step 7: Initialize the trainer
|