Spaces:
Running
Running
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@
|
|
|
|
|
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@
|
|
|
|
|
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 |
}
|