Jayant-Kernel commited on
Commit ·
354d3fd
1
Parent(s): 88fb03e
fix: add matplotlib, split COPY into separate lines
Browse files- Dockerfile +3 -2
Dockerfile
CHANGED
|
@@ -8,7 +8,7 @@ RUN apt-get update && apt-get install -y git build-essential && rm -rf /var/lib/
|
|
| 8 |
|
| 9 |
WORKDIR /app
|
| 10 |
|
| 11 |
-
RUN pip install --no-cache-dir torch transformers peft trl bitsandbytes accelerate wandb datasets huggingface_hub
|
| 12 |
|
| 13 |
RUN pip install --no-cache-dir git+https://github.com/Jayant-kernel/DECEIT-the-ai-truth-environment-.git
|
| 14 |
|
|
@@ -19,6 +19,7 @@ RUN mkdir -p /usr/local/lib/python3.10/site-packages/deceit_env/data/ && \
|
|
| 19 |
COPY data/ /usr/local/lib/python3.10/site-packages/deceit_env/data/
|
| 20 |
COPY data/ /home/trainer/.local/lib/python3.10/site-packages/deceit_env/data/
|
| 21 |
COPY data/ /app/data/
|
| 22 |
-
COPY train.py
|
|
|
|
| 23 |
|
| 24 |
CMD ["python", "evaluate.py"]
|
|
|
|
| 8 |
|
| 9 |
WORKDIR /app
|
| 10 |
|
| 11 |
+
RUN pip install --no-cache-dir torch transformers peft trl bitsandbytes accelerate wandb datasets huggingface_hub matplotlib Pillow
|
| 12 |
|
| 13 |
RUN pip install --no-cache-dir git+https://github.com/Jayant-kernel/DECEIT-the-ai-truth-environment-.git
|
| 14 |
|
|
|
|
| 19 |
COPY data/ /usr/local/lib/python3.10/site-packages/deceit_env/data/
|
| 20 |
COPY data/ /home/trainer/.local/lib/python3.10/site-packages/deceit_env/data/
|
| 21 |
COPY data/ /app/data/
|
| 22 |
+
COPY train.py .
|
| 23 |
+
COPY evaluate.py .
|
| 24 |
|
| 25 |
CMD ["python", "evaluate.py"]
|