File size: 679 Bytes
9e64e71 | 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 27 28 29 | FROM python:3.12-slim
WORKDIR /app
# Install git for pip installs from GitHub
RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*
# Copy project
COPY . .
# Install sql-env without deps
RUN pip install --no-deps .
# Install training deps (same as Colab setup cell)
RUN pip install \
"trl>=0.29.0" \
"accelerate>=0.34.0" \
"openenv-core[core]>=0.2.1" \
"pydantic>=2.0.0" \
"jmespath" \
"datasets>=3.0.0" \
"huggingface_hub>=0.30.0" \
"git+https://github.com/huggingface/transformers.git@main"
# Download Spider databases
RUN python scripts/download_spider_databases.py
CMD ["python", "scripts/test_training_local.py"]
|