Hugging Face's logo

Spaces:
daresearch
/
train_70b_4bit
Runtime error

App Files Community
daresearch commited on
Commit
2987771
·
verified ·
1 Parent(s): 91e3659

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -66,4 +66,4 @@ RUN mkdir -p /root/.cache/huggingface/accelerate && \
66
  > /root/.cache/huggingface/accelerate/default_config.yaml
67
 
68
  # Launch your training script on 4 GPUs with fp16
69
- CMD ["accelerate", "launch", "--num_processes=1", "--mixed_precision=fp16", "finetune_script.py"]
 
66
  > /root/.cache/huggingface/accelerate/default_config.yaml
67
 
68
  # Launch your training script on 4 GPUs with fp16
69
+ CMD ["accelerate", "launch", "--num_processes=4", "--mixed_precision=fp16", "finetune_script.py"]