File size: 903 Bytes
00e4b4e 5552679 00e4b4e 5552679 00e4b4e 5552679 00e4b4e cba6bf8 | 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 30 31 32 33 34 35 36 37 38 39 | # FROM argilla/argilla-hf-spaces:v2.8.0
# Copy the auth config section
# COPY .oauth.yaml /home/argilla/
# Comment this line to disable annotation progress sharing feature
# ENV ARGILLA_ENABLE_SHARE_YOUR_PROGRESS=1
# Uncoment this line to remove the persistence storage warning
#ENV ARGILLA_SHOW_HUGGINGFACE_SPACE_PERSISTENT_STORAGE_WARNING=false
FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-runtime
# Install basic tools
RUN apt-get update && apt-get install -y git curl
# Install Python packages
RUN pip install --upgrade pip
RUN pip install \
datasets \
transformers \
accelerate \
peft \
trl \
spacy \
chromadb \
huggingface_hub \
tensorboard
# Download spaCy model
RUN python -m spacy download en_core_web_sm
# Copy all files into the container
COPY . /workspace
WORKDIR /workspace
# Set the command to run your training script
CMD ["python", "train.py"]
|