Update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
|
@@ -1,2 +1,2 @@
|
|
| 1 |
FROM huggingface/autotrain-advanced:latest
|
| 2 |
-
CMD pip install
|
|
|
|
| 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
|