| | |
| | |
| | |
| | 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"] |
| |
|