Upload train.py with huggingface_hub
Browse files
train.py
CHANGED
|
@@ -117,7 +117,6 @@ def main():
|
|
| 117 |
report_to="none",
|
| 118 |
warmup_steps=100,
|
| 119 |
lr_scheduler_type="cosine",
|
| 120 |
-
max_seq_length=1024,
|
| 121 |
# Push to Hugging Face Hub
|
| 122 |
push_to_hub=True,
|
| 123 |
hub_model_id=f"{HF_USERNAME}/{OUTPUT_DIR}",
|
|
|
|
| 117 |
report_to="none",
|
| 118 |
warmup_steps=100,
|
| 119 |
lr_scheduler_type="cosine",
|
|
|
|
| 120 |
# Push to Hugging Face Hub
|
| 121 |
push_to_hub=True,
|
| 122 |
hub_model_id=f"{HF_USERNAME}/{OUTPUT_DIR}",
|