Hugging Face's logo

Spaces:
nroggendorff
/
train-llama
Runtime error

App Files Community
34

Merge when ready d'MFmJGh

#28
by nroggendorff - opened
base: refs/heads/main
←
from: refs/pr/28
Discussion Files changed
+1
-1
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -6,4 +6,4 @@ RUN tmp=$(mktemp) && \
6
  mv "$tmp" "${CONFIG}"
7
  RUN chown -R user:user config.json
8
 
9
- CMD ["bash", "-c", "python prep.py && deepspeed --num_gpus 8 train.py"]
 
6
  mv "$tmp" "${CONFIG}"
7
  RUN chown -R user:user config.json
8
 
9
+ CMD ["bash", "-c", "python prep.py && deepspeed train.py"]