Spaces:
Runtime error
Runtime error
Update fine_tune.py
Browse files- fine_tune.py +2 -0
fine_tune.py
CHANGED
|
@@ -69,6 +69,8 @@ if not os.path.exists(os.path.join(output_dir, 'adapter_config.json')):
|
|
| 69 |
warmup_ratio=0.03,
|
| 70 |
group_by_length=True,
|
| 71 |
lr_scheduler_type="linear",
|
|
|
|
|
|
|
| 72 |
)
|
| 73 |
|
| 74 |
# Initialize Trainer
|
|
|
|
| 69 |
warmup_ratio=0.03,
|
| 70 |
group_by_length=True,
|
| 71 |
lr_scheduler_type="linear",
|
| 72 |
+
push_to_hub=True,
|
| 73 |
+
hub_model_id="Nutnell/DirectEd-AI",
|
| 74 |
)
|
| 75 |
|
| 76 |
# Initialize Trainer
|