Start work on llm_engine

This commit is contained in:
2024-10-20 19:05:47 +02:00
parent 8114e98bce
commit 006db518f2
8 changed files with 195 additions and 0 deletions

12
Dockerfile Normal file
View 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