onnx export

FlagAI-Open · Jul 30, 2023 · eb58f2b · eb58f2b
1 parent 427cdf2
commit eb58f2b
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/onnx/Dockerfile b/onnx/Dockerfile
@@ -25,10 +25,20 @@ RUN git clone --depth=1 $GIT
 
 ADD os/ /
 
+RUN pip install -i https://mirrors.aliyun.com/pypi/simple/ PyYAML scikit-learn Pillow
+
 RUN cd FlagAI &&\
+  sed -i '/PyYAML/d' setup.py &&\
+  sed -i '/scikit-learn/d' setup.py &&\
+  sed -i '/Pillow/d' setup.py &&\
   rm setup.cfg &&\
   python setup.py install
 
 RUN pip uninstall -y google-auth 
 
+RUN \
+  pip uninstall -y protobuf &&\
+  pip install  -i https://mirrors.aliyun.com/pypi/simple/ protobuf
+
+RUN pip install  -i https://mirrors.aliyun.com/pypi/simple/ onnx
 ENV PYTHONPATH /app