Spaces:
No application file
No application file
File size: 351 Bytes
b7edc7f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
FROM cnstark/pytorch:2.0.1-py3.9.17-cuda11.8.0-ubuntu20.04
WORKDIR /app
COPY requirements.txt /app/
RUN pip install -r requirements.txt && \
pip install tiktoken && \
pip install transformers_stream_generator
COPY . /app/
VOLUME [ "/root/.cache/huggingface/", "/app/data", "/app/output" ]
EXPOSE 7860
CMD [ "python", "src/train_web.py" ]
|