BloodJackson commited on
Commit
669575f
·
verified ·
1 Parent(s): 6902f83

Update Dockerfile

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