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

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

WORKDIR /opt/runpod

COPY templates/video-finetuner/workspace/finetune.sh /opt/runpod/bin/finetune.sh

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

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