YashashMathur commited on
Commit
ce8daf0
·
verified ·
1 Parent(s): c51cef7

Upload Dockerfile with huggingface_hub

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -33,7 +33,7 @@ RUN python3 -m pip install --no-cache-dir torch==2.4.0 torchvision==0.19.0 torch
33
  RUN python3 -m pip install --no-cache-dir "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git"
34
 
35
  # 4. Install HF Stack normally (letting pip resolve dependencies)
36
- RUN python3 -m pip install --no-cache-dir transformers==4.45.0 peft==0.13.0 trl==0.11.0 accelerate bitsandbytes datasets huggingface_hub gradio scipy sentencepiece protobuf safetensors
37
 
38
  # 5. Surgical strike against torchao
39
  RUN python3 -m pip uninstall -y torchao
 
33
  RUN python3 -m pip install --no-cache-dir "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git"
34
 
35
  # 4. Install HF Stack normally (letting pip resolve dependencies)
36
+ RUN python3 -m pip install --no-cache-dir transformers peft trl accelerate bitsandbytes datasets huggingface_hub gradio scipy sentencepiece protobuf safetensors
37
 
38
  # 5. Surgical strike against torchao
39
  RUN python3 -m pip uninstall -y torchao