Start work on llm_engine
This commit is contained in:
12
Dockerfile
Normal file
12
Dockerfile
Normal file
@@ -0,0 +1,12 @@
|
||||
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
|
||||
Reference in New Issue
Block a user