Jayant-Kernel commited on
Commit
354d3fd
·
1 Parent(s): 88fb03e

fix: add matplotlib, split COPY into separate lines

Browse files
Files changed (1) hide show
  1. 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 evaluate.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"]