12 lines
313 B
Docker
12 lines
313 B
Docker
FROM huggingface/transformers-pytorch-gpu AS requirements
|
|
WORKDIR /root
|
|
COPY requirements.txt /root/requirements.txt
|
|
COPY ./sia/ /root/sia/
|
|
RUN pip3 install -r requirements.txt
|
|
|
|
FROM requirements AS test
|
|
COPY ./tests/ /root/tests/
|
|
RUN mkdir -p /root/model
|
|
CMD python3 -m unittest discover tests
|
|
|
|
FROM requirements |