betterwithage commited on
Commit
d0fe8e4
·
verified ·
1 Parent(s): c3e05e1

WILLAY: drop trackio (parquet hook bug), push clean

Browse files
Files changed (1) hide show
  1. 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", "trackio", "accelerate>=0.34.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="trackio",
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()