Spaces:
Runtime error
Runtime error
Julian Bilcke
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@
|
|
|
|
|
|
|
| 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 |
}
|