Update Dockerfile
Browse files- Dockerfile +3 -3
Dockerfile
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
FROM huggingface/autotrain-advanced:latest
|
| 2 |
-
|
| 3 |
-
|
| 4 |
-
|
| 5 |
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 |
+
RUN pip install -r requirements.txt
|
| 3 |
+
RUN pip install accelerate
|
| 4 |
+
RUN pip install -i https://pypi.org/simple/bitsandbytes
|
| 5 |
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
|