fnruha0921 commited on
Commit
b45d7df
·
verified ·
1 Parent(s): e464081

Downgrade bitsandbytes==0.43.3 to fix peft 0.14.0 dispatch_bnb_4bit compress_statistics attr

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -0
Dockerfile CHANGED
@@ -3,4 +3,5 @@ CMD pip uninstall -y autotrain-advanced && \
3
  pip install -U autotrain-advanced && \
4
  pip install --upgrade --no-deps "transformers==4.51.3" && \
5
  pip install --upgrade --no-deps "huggingface-hub>=0.30.0,<1.0" && \
 
6
  autotrain app --host 0.0.0.0 --port 7860 --workers 1
 
3
  pip install -U autotrain-advanced && \
4
  pip install --upgrade --no-deps "transformers==4.51.3" && \
5
  pip install --upgrade --no-deps "huggingface-hub>=0.30.0,<1.0" && \
6
+ pip install --upgrade --no-deps "bitsandbytes==0.43.3" && \
7
  autotrain app --host 0.0.0.0 --port 7860 --workers 1