nroggendorff commited on
Commit
954d26f
1 Parent(s): f74ff73

Merge when ready 03|\sD4e

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -6
Dockerfile CHANGED
@@ -1,9 +1,7 @@
1
  FROM nroggendorff/train-llama:latest
2
 
3
- ARG CONFIG="config.json"
4
- RUN tmp=$(mktemp) && \
5
- jq '.init = 0 | ."instruct-finetune-bool" = false' "${CONFIG}" > "$tmp" && \
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"]