Veinnn commited on
Commit
a220aad
·
verified ·
1 Parent(s): 7a4c38b

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -0
Dockerfile CHANGED
@@ -1,2 +1,4 @@
1
  FROM huggingface/autotrain-advanced:latest
2
  CMD pip uninstall -y autotrain-advanced && pip install -U autotrain-advanced && autotrain app --host 0.0.0.0 --port 7860 --workers 1
 
 
 
1
  FROM huggingface/autotrain-advanced:latest
2
  CMD pip uninstall -y autotrain-advanced && pip install -U autotrain-advanced && autotrain app --host 0.0.0.0 --port 7860 --workers 1
3
+ CMD pip install backoff
4
+ CMD pip install flash-attn --no-build-isolation