init commit

This commit is contained in:
Jinx
2026-04-10 06:24:05 +00:00
parent 46a3ddbb25
commit adbd85366b
3 changed files with 101 additions and 294 deletions

View File

@@ -2,18 +2,19 @@ FROM pytorch/pytorch:2.5.1-cuda12.4-cudnn9-devel
# System deps
RUN apt-get update && apt-get install -y --no-install-recommends \
git ninja-build packaging wget curl \
git ninja-build wget curl \
&& rm -rf /var/lib/apt/lists/*
# Python deps
COPY requirements.txt /tmp/requirements.txt
RUN pip install --no-cache-dir -r /tmp/requirements.txt
RUN pip install --no-cache-dir -r /tmp/requirements.txt packaging
# Copy scripts
WORKDIR /app
COPY prepare_data.py /app/
COPY train_lora.py /app/
COPY run.sh /app/
RUN chmod +x /app/run.sh
# Data and output dirs
RUN mkdir -p /data/processed /data/lora-output /data/models