Jayant-Kernel commited on
fix: add unsloth_zoo dependency
Browse files- 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 .
|