Jayant-Kernel commited on
Commit
0db5a23
·
unverified ·
1 Parent(s): 8fb443c

fix: install unsloth_zoo in Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -3,7 +3,7 @@ RUN apt-get update && apt-get install -y git build-essential && rm -rf /var/lib/
3
  WORKDIR /app
4
  COPY evaluate.py .
5
  RUN pip install -q torch transformers huggingface_hub matplotlib
6
- RUN pip install "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git" --no-deps
7
  RUN pip install trl peft accelerate bitsandbytes datasets
8
  RUN pip install git+https://github.com/Jayant-kernel/DECEIT-the-ai-truth-environment-.git
9
  ENV PYTHONUNBUFFERED=1
 
3
  WORKDIR /app
4
  COPY evaluate.py .
5
  RUN pip install -q torch transformers huggingface_hub matplotlib
6
+ RUN pip install unsloth unsloth_zoo nest-asyncio
7
  RUN pip install trl peft accelerate bitsandbytes datasets
8
  RUN pip install git+https://github.com/Jayant-kernel/DECEIT-the-ai-truth-environment-.git
9
  ENV PYTHONUNBUFFERED=1