jbilcke-hf HF staff commited on
Commit
0328e32
·
1 Parent(s): f1c60d3

tmp fix for finetrainers

Browse files
requirements.txt CHANGED
@@ -28,7 +28,9 @@ pynvml
28
  #eva-decord==0.6.1
29
  decord
30
 
31
- finetrainers @ git+https://github.com/a-r-r-o-w/finetrainers.git@4c6844fabf496fb622b6333f57e8ebecee6f2780
 
 
32
 
33
  wandb
34
  pandas
 
28
  #eva-decord==0.6.1
29
  decord
30
 
31
+ #finetrainers @ git+https://github.com/a-r-r-o-w/finetrainers.git@main
32
+ # temporary fix for pip install bug:
33
+ finetrainers @ git+https://github.com/jbilcke-hf/finetrainers-patches.git@fix_missing_sft_trainer_files
34
 
35
  wandb
36
  pandas
requirements_without_flash_attention.txt CHANGED
@@ -22,14 +22,16 @@ bitsandbytes
22
  peft>=0.12.0
23
 
24
  # For GPU monitoring of NVIDIA chipsets
25
- you probably won't be able to install that on macOS
26
  # pynvml
27
 
28
  # eva-decord is missing get_batch it seems
29
  eva-decord==0.6.1
30
  # decord
31
 
32
- finetrainers @ git+https://github.com/a-r-r-o-w/finetrainers.git@4c6844fabf496fb622b6333f57e8ebecee6f2780
 
 
33
 
34
  wandb
35
  pandas
 
22
  peft>=0.12.0
23
 
24
  # For GPU monitoring of NVIDIA chipsets
25
+ # you probably won't be able to install that on macOS
26
  # pynvml
27
 
28
  # eva-decord is missing get_batch it seems
29
  eva-decord==0.6.1
30
  # decord
31
 
32
+ #finetrainers @ git+https://github.com/a-r-r-o-w/finetrainers.git@main
33
+ # temporary fix for pip install bug:
34
+ finetrainers @ git+https://github.com/jbilcke-hf/finetrainers-patches.git@fix_missing_sft_trainer_files
35
 
36
  wandb
37
  pandas
vms/ui/project/services/training.py CHANGED
@@ -1078,7 +1078,7 @@ class TrainingService:
1078
  "learning_rate": ui_state.get("learning_rate", DEFAULT_LEARNING_RATE),
1079
  "save_iterations": ui_state.get("save_iterations", DEFAULT_SAVE_CHECKPOINT_EVERY_N_STEPS),
1080
  "preset_name": ui_state.get("training_preset", list(TRAINING_PRESETS.keys())[0]),
1081
- "repo_id": "" # Default empty repo ID,
1082
  "auto_resume": ui_state.get("auto_resume", DEFAULT_AUTO_RESUME)
1083
  }
1084
  }
 
1078
  "learning_rate": ui_state.get("learning_rate", DEFAULT_LEARNING_RATE),
1079
  "save_iterations": ui_state.get("save_iterations", DEFAULT_SAVE_CHECKPOINT_EVERY_N_STEPS),
1080
  "preset_name": ui_state.get("training_preset", list(TRAINING_PRESETS.keys())[0]),
1081
+ "repo_id": "", # Default empty repo ID,
1082
  "auto_resume": ui_state.get("auto_resume", DEFAULT_AUTO_RESUME)
1083
  }
1084
  }