betterwithage commited on
Commit
51608f0
·
verified ·
1 Parent(s): 228f5fb

fix(train): report_to none so Trackio 404 cannot poison adapter upload

Browse files
Files changed (1) hide show
  1. train_khipu_abstain.py +2 -3
train_khipu_abstain.py CHANGED
@@ -482,13 +482,12 @@ def main() -> None:
482
  lr_scheduler_type="constant_with_warmup",
483
  seed=SEED,
484
  output_dir="outputs",
485
- report_to="trackio",
486
- run_name="khipu-abstain-oversample4",
487
  save_strategy="no",
488
  push_to_hub=False,
489
  )
490
  try:
491
- args = SFTConfig(**sft_kwargs, project="szl-khipu-abstain")
492
  except TypeError:
493
  args = SFTConfig(**sft_kwargs)
494
 
 
482
  lr_scheduler_type="constant_with_warmup",
483
  seed=SEED,
484
  output_dir="outputs",
485
+ report_to="none",
 
486
  save_strategy="no",
487
  push_to_hub=False,
488
  )
489
  try:
490
+ args = SFTConfig(**sft_kwargs)
491
  except TypeError:
492
  args = SFTConfig(**sft_kwargs)
493