Siddh12334 commited on
Commit
1a15203
·
verified ·
1 Parent(s): fbb358f

fix: install unsloth in Dockerfile at build time (not at click-time)

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -2
Dockerfile CHANGED
@@ -8,9 +8,10 @@ RUN apt-get update && apt-get install -y git build-essential && rm -rf /var/lib/
8
  RUN pip install --no-cache-dir \
9
  torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 && \
10
  pip install --no-cache-dir \
11
- trl transformers datasets accelerate \
12
  openenv-core fastapi uvicorn pydantic \
13
- wandb faker python-dotenv gradio
 
14
 
15
  COPY . .
16
 
 
8
  RUN pip install --no-cache-dir \
9
  torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 && \
10
  pip install --no-cache-dir \
11
+ trl transformers datasets accelerate peft bitsandbytes \
12
  openenv-core fastapi uvicorn pydantic \
13
+ wandb faker python-dotenv gradio && \
14
+ pip install --no-cache-dir "unsloth @ git+https://github.com/unslothai/unsloth.git"
15
 
16
  COPY . .
17