Spaces:
Build error
Build error
Update train.py
Browse files
train.py
CHANGED
@@ -58,7 +58,7 @@ training_args = TrainingArguments(
|
|
58 |
save_steps=50,
|
59 |
fp16=True, # Use mixed precision
|
60 |
push_to_hub=True,
|
61 |
-
hub_model_id="lyricolivia20/LivLoRA
|
62 |
)
|
63 |
|
64 |
# 8. Create trainer
|
|
|
58 |
save_steps=50,
|
59 |
fp16=True, # Use mixed precision
|
60 |
push_to_hub=True,
|
61 |
+
hub_model_id="lyricolivia20/LivLoRA", # Your HF repo
|
62 |
)
|
63 |
|
64 |
# 8. Create trainer
|