WILLAY: drop trackio (parquet hook bug), push clean
Browse files- train_willay.py +2 -4
train_willay.py
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
# /// script
|
| 2 |
-
# dependencies = ["trl>=0.12.0", "peft>=0.7.0", "transformers>=4.44.0", "datasets>=2.20.0", "
|
| 3 |
# ///
|
| 4 |
"""
|
| 5 |
WILLAY — signed-refusal specialist for SZL Holdings.
|
|
@@ -101,9 +101,7 @@ def main():
|
|
| 101 |
logging_steps=5,
|
| 102 |
save_strategy="no",
|
| 103 |
max_length=1024,
|
| 104 |
-
report_to="
|
| 105 |
-
project="szl-willay",
|
| 106 |
-
run_name="willay-refusal-specialist-v1",
|
| 107 |
),
|
| 108 |
)
|
| 109 |
trainer.train()
|
|
|
|
| 1 |
# /// script
|
| 2 |
+
# dependencies = ["trl>=0.12.0", "peft>=0.7.0", "transformers>=4.44.0", "datasets>=2.20.0", "accelerate>=0.34.0"]
|
| 3 |
# ///
|
| 4 |
"""
|
| 5 |
WILLAY — signed-refusal specialist for SZL Holdings.
|
|
|
|
| 101 |
logging_steps=5,
|
| 102 |
save_strategy="no",
|
| 103 |
max_length=1024,
|
| 104 |
+
report_to="none",
|
|
|
|
|
|
|
| 105 |
),
|
| 106 |
)
|
| 107 |
trainer.train()
|