YashashMathur commited on
Commit
f5ca8fb
·
verified ·
1 Parent(s): e5115bd

Upload Dockerfile with huggingface_hub

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -0
Dockerfile CHANGED
@@ -38,6 +38,9 @@ RUN python3 -m pip install unsloth_zoo
38
  # e. pip install --no-deps xformers==0.0.27 trl==0.11.0 peft accelerate bitsandbytes
39
  RUN python3 -m pip install --no-deps xformers==0.0.27 trl==0.11.0 peft accelerate bitsandbytes
40
 
 
 
 
41
  # Install other requirements
42
  COPY --chown=user:user requirements.txt .
43
  RUN python3 -m pip install -r requirements.txt
 
38
  # e. pip install --no-deps xformers==0.0.27 trl==0.11.0 peft accelerate bitsandbytes
39
  RUN python3 -m pip install --no-deps xformers==0.0.27 trl==0.11.0 peft accelerate bitsandbytes
40
 
41
+ # f. Fix transformers version (unsloth_zoo needs specific version - 4.46.x works)
42
+ RUN python3 -m pip install transformers==4.46.3
43
+
44
  # Install other requirements
45
  COPY --chown=user:user requirements.txt .
46
  RUN python3 -m pip install -r requirements.txt