Spaces:
Build error
Build error
Update tune-lf-4gpu.sh
Browse files- scripts/tune-lf-4gpu.sh +1 -1
scripts/tune-lf-4gpu.sh
CHANGED
|
@@ -12,4 +12,4 @@ export DATA_PATH=../datasets/mac/mac.tsv
|
|
| 12 |
export YAML=config/mac_template_4gpu.yaml
|
| 13 |
|
| 14 |
python ../llm_toolkit/setup_lf.py
|
| 15 |
-
llamafactory-cli train config/models/$MODEL_NAME.yaml
|
|
|
|
| 12 |
export YAML=config/mac_template_4gpu.yaml
|
| 13 |
|
| 14 |
python ../llm_toolkit/setup_lf.py
|
| 15 |
+
CUDA_VISIBLE_DEVICES=0,1,2,3 llamafactory-cli train config/models/$MODEL_NAME.yaml
|