anezatra2 commited on
Commit
3ec4d18
·
verified ·
1 Parent(s): 9672ca8

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -1,2 +1,2 @@
1
  FROM huggingface/autotrain-advanced:latest
2
- CMD pip install accelerate --upgrade && pip install bitsandbytes --upgrade && pip uninstall -y autotrain-advanced && pip install -U autotrain-advanced && uvicorn autotrain.app:app --host 0.0.0.0 --port 7860 --reload --workers 4
 
1
  FROM huggingface/autotrain-advanced:latest
2
+ CMD pip install -i https://pypi.org/simple/ bitsandbytes && pip install bitsandbytes-cuda111 && pip install accelerate --upgrade && pip uninstall -y autotrain-advanced && pip install -U autotrain-advanced && uvicorn autotrain.app:app --host 0.0.0.0 --port 7860 --reload --workers 4