Nutnell commited on
Commit
f2dc681
·
verified ·
1 Parent(s): 47dca3b

Update fine_tune.py

Browse files
Files changed (1) hide show
  1. 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