YashashMathur commited on
Commit
da61617
·
verified ·
1 Parent(s): 8b092ea

Upload Dockerfile with huggingface_hub

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -0
Dockerfile CHANGED
@@ -32,6 +32,9 @@ RUN python3 -m pip uninstall -y torchao
32
  # d. pip install --no-deps "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git"
33
  RUN python3 -m pip install --no-deps "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git"
34
 
 
 
 
35
  # e. pip install --no-deps xformers==0.0.27 trl==0.11.0 peft accelerate bitsandbytes
36
  RUN python3 -m pip install --no-deps xformers==0.0.27 trl==0.11.0 peft accelerate bitsandbytes
37
 
 
32
  # d. pip install --no-deps "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git"
33
  RUN python3 -m pip install --no-deps "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git"
34
 
35
+ # d.1 Install unsloth_zoo (required dependency for unsloth to work)
36
+ RUN python3 -m pip install unsloth_zoo
37
+
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