From eb58f2b6ac3c91c0ef7208a1b0bed016e62ae440 Mon Sep 17 00:00:00 2001
From: "xxai.art" <xxai.art@gmail.com>
Date: Sun, 30 Jul 2023 14:49:16 +0800
Subject: [PATCH] onnx export

---
 onnx/Dockerfile | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/onnx/Dockerfile b/onnx/Dockerfile
index f3a71406..4cd73bf3 100755
--- 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