prodigy-rlhf-ranking / Dockerfile
wesslen's picture
add rlhf_ranking
9bb55ac
raw
history blame
972 Bytes
FROM python:3.9
#COPY requirements.txt /app/
WORKDIR /app
# # Set up a new user named "user" with user ID 1000
# RUN useradd -m -u 1000 user
# # Switch to the "user" user
# USER user
# # Set home to the user's home directory
# ENV HOME=/home/user \
# PATH=/home/user/.local/bin:$PATH
# # Set the working directory to the user's home directory
# WORKDIR $HOME/app
# # Copy the current directory contents into the container at $HOME/app setting the owner to the user
# COPY --chown=user . $HOME/app
RUN --mount=type=secret,id=LICENSE_KEY,mode=0444,required=true \
pip install --upgrade pip \
&& pip install prodigy -f https://$(cat /run/secrets/LICENSE_KEY)@download.prodi.gy \
&& python -m spacy download en_core_web_sm
RUN chmod 777 .
COPY prodigy.json .
COPY data ./data/
COPY rlhf-ranking.py .
COPY prodigy.sh .
ENV PRODIGY_HOME /app
ENV PRODIGY_LOGGING "verbose"
ENV PRODIGY_ALLOWED_SESSIONS "user1,user2"
EXPOSE 7860
CMD CMD ["bash","prodigy.sh"]