File size: 855 Bytes
1058c6b 3470129 1670c46 237e430 fbc565d cbaf9f7 1058c6b 3470129 e30d685 0efac4a 787e377 e30d685 54fc539 11baf5d e30d685 a0058bb 354d3fd 4838647 e30d685 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 | FROM python:3.10-slim
ENV PYTHONUNBUFFERED=1
ENV HF_HOME=/tmp/huggingface
ENV HOME=/tmp
RUN apt-get update && apt-get install -y git build-essential && rm -rf /var/lib/apt/lists/*
WORKDIR /app
RUN pip install --no-cache-dir torch transformers peft trl bitsandbytes accelerate wandb datasets huggingface_hub matplotlib Pillow
RUN pip install --no-cache-dir git+https://github.com/Jayant-kernel/DECEIT-the-ai-truth-environment-.git
RUN mkdir -p /usr/local/lib/python3.10/site-packages/deceit_env/data/ && \
mkdir -p /home/trainer/.local/lib/python3.10/site-packages/deceit_env/data/ && \
mkdir -p /app/data/
COPY data/ /usr/local/lib/python3.10/site-packages/deceit_env/data/
COPY data/ /home/trainer/.local/lib/python3.10/site-packages/deceit_env/data/
COPY data/ /app/data/
COPY train.py .
COPY evaluate.py .
CMD ["python", "evaluate.py"]
|