Spaces:
Runtime error
Runtime error
nroggendorff
commited on
Commit
•
954d26f
1
Parent(s):
f74ff73
Merge when ready 03|\sD4e
Browse files- Dockerfile +4 -6
Dockerfile
CHANGED
@@ -1,9 +1,7 @@
|
|
1 |
FROM nroggendorff/train-llama:latest
|
2 |
|
3 |
-
|
4 |
-
|
5 |
-
|
6 |
-
mv "$tmp" "${CONFIG}"
|
7 |
-
RUN chown -R user:user config.json
|
8 |
|
9 |
-
CMD ["bash", "-c", "python prep.py && deepspeed --num_gpus=$(nvidia-smi --query-gpu=gpu_name --format=csv,noheader | wc -l) train.py"]
|
|
|
1 |
FROM nroggendorff/train-llama:latest
|
2 |
|
3 |
+
RUN jq '.init = 0 | ."instruct-finetune-bool" = false' config.json > temp.json && \
|
4 |
+
mv temp.json config.json && \
|
5 |
+
chown -R user:user config.json
|
|
|
|
|
6 |
|
7 |
+
CMD ["bash", "-c", "python prep.py && deepspeed --num_gpus=$(nvidia-smi --query-gpu=gpu_name --format=csv,noheader | wc -l) train.py"]
|