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