train / Dockerfile
qgallouedec's picture
qgallouedec HF Staff
Update Dockerfile
03966ae verified
raw
history blame
663 Bytes
# Use a lightweight Python image as the base
FROM python:3-slim
# Set the working directory inside the container
WORKDIR /app
# Install git and clean up to keep the image small
RUN apt-get update && \
apt-get install -y git && \
apt-get clean && \
rm -rf /var/lib/apt/lists/*
# Copy all files to the container
COPY . /app
# Install required dependencies
RUN pip install --upgrade pip
RUN pip install -r requirements.txt
# Copy the shell script and make it executable
COPY sft.sh /usr/local/bin/sft
RUN chmod +x /usr/local/bin/sft
# Optional: set entrypoint to sft
ENTRYPOINT ["sft"]
# Default command to run the script
CMD ["python", "main.py"]