|
|
|
|
|
|
|
FROM python:3.11 |
|
|
|
|
|
WORKDIR /gpt |
|
|
|
|
|
COPY . . |
|
|
|
|
|
RUN pip3 install -r requirements.txt |
|
|
|
|
|
RUN apt-get update && apt-get install libgl1 -y |
|
RUN pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cpu |
|
RUN pip3 install transformers protobuf langchain sentence-transformers faiss-cpu nltk beautifulsoup4 bitsandbytes tabulate icetk --upgrade |
|
RUN pip3 install unstructured[all-docs] --upgrade |
|
RUN python3 -c 'from check_proxy import warm_up_vectordb; warm_up_vectordb()' |
|
|
|
|
|
RUN python3 -c 'from check_proxy import warm_up_modules; warm_up_modules()' |
|
|
|
|
|
CMD ["python3", "-u", "main.py"] |
|
|