FROM runpod/pytorch:1.0.2-cu1281-torch280-ubuntu2404

ENV PYTHONUNBUFFERED=1 \
    PATH="/opt/runpod/venv/bin:${PATH}"

WORKDIR /opt/runpod

COPY templates/lora-trainer/workspace/train.sh /opt/runpod/bin/train.sh

RUN chmod +x /opt/runpod/bin/train.sh \
    && python3 -m venv /opt/runpod/venv \
    && . /opt/runpod/venv/bin/activate \
    && pip install --upgrade pip

# Install your training stack here, for example:
# RUN . /opt/runpod/venv/bin/activate \
#     && pip install unsloth peft trl accelerate bitsandbytes