Hugging Face's logo

Spaces:
nroggendorff
/
train-llama
Runtime error

App Files Community
34
nroggendorff commited on
Commit
3050d31
•
1 Parent(s): a47602a

Update Dockerfile

Browse files
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 train.py"]
 
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"]