diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000..79c029e --- /dev/null +++ b/Dockerfile @@ -0,0 +1,34 @@ +FROM python:3.8 + +MAINTAINER "chatGLM" + +COPY agent /chatGLM/agent + +COPY chains /chatGLM/chains + +COPY configs /chatGLM/configs + +COPY content /chatGLM/content + +COPY models /chatGLM/models + +COPY nltk_data /chatGLM/content + +COPY requirements.txt /chatGLM/ + +COPY cli_demo.py /chatGLM/ + +COPY webui.py /chatGLM/ + +WORKDIR /chatGLM + +RUN pip install --user torch torchvision tensorboard cython -i https://pypi.tuna.tsinghua.edu.cn/simple +# RUN pip install --user 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI' + +# RUN pip install --user 'git+https://github.com/facebookresearch/fvcore' +# install detectron2 +# RUN git clone https://github.com/facebookresearch/detectron2 + +RUN pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple/ --trusted-host pypi.tuna.tsinghua.edu.cn + +CMD ["python","-u", "webui.py"]