Jayant-Kernel commited on
Commit
4550c9f
·
unverified ·
1 Parent(s): fbc565d

fix: add unsloth_zoo dependency

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -0
Dockerfile CHANGED
@@ -16,6 +16,8 @@ RUN pip install --no-cache-dir trl peft accelerate bitsandbytes wandb datasets h
16
 
17
  RUN pip install --no-cache-dir "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git" --no-deps
18
 
 
 
19
  RUN pip install --no-cache-dir git+https://github.com/Jayant-kernel/DECEIT-the-ai-truth-environment-.git
20
 
21
  COPY train.py .
 
16
 
17
  RUN pip install --no-cache-dir "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git" --no-deps
18
 
19
+ RUN pip install --no-cache-dir unsloth_zoo
20
+
21
  RUN pip install --no-cache-dir git+https://github.com/Jayant-kernel/DECEIT-the-ai-truth-environment-.git
22
 
23
  COPY train.py .