feat: upload onnx executor

jina-ai · Jun 14, 2022 · e449a78 · e449a78
1 parent f8b5eca
commit e449a78
Showing 1 changed file with 7 additions and 9 deletions.
diff --git a/.github/workflows/force-hub-push.yml b/.github/workflows/force-hub-push.yml
@@ -145,18 +145,16 @@ jobs:
           
           python scripts/get-requirements.py onnx server/requirements.txt
           
-          cp Dockerfiles/base.Dockerfile server/Dockerfile
-          cp Dockerfiles/cuda.Dockerfile server/Dockerfile.gpu
-          
-          sed -i 's/ARG BACKEND_TAG=torch/ARG BACKEND_TAG=onnx/g' server/Dockerfile
-          sed -i 's/ARG BACKEND_TAG=torch/ARG BACKEND_TAG=onnx/g' server/Dockerfile.gpu
-          
           exec_name=`yq -r .name server/manifest.yml`
           echo executor name is $exec_name
           
-          # FIXME: disable uploading at debugging
-          # JINA_AUTH_TOKEN=${{secrets.JINAHUB_TOKEN}} jina hub push --force $exec_name --secret ${{secrets.ONNX_EXEC_SECRET}} server ${{env.TAG_ALIAS}}
-
+          cp Dockerfiles/base.Dockerfile server/Dockerfile
+          sed -i 's/ARG BACKEND_TAG=torch/ARG BACKEND_TAG=onnx/g' server/Dockerfile          
+          JINA_AUTH_TOKEN=${{secrets.JINAHUB_TOKEN}} jina hub push --force $exec_name --secret ${{secrets.ONNX_EXEC_SECRET}} server ${{env.TAG_ALIAS}}
+          
+          cp Dockerfiles/cuda.Dockerfile server/Dockerfile
+          sed -i 's/ARG BACKEND_TAG=torch/ARG BACKEND_TAG=onnx/g' server/Dockerfile
+          JINA_AUTH_TOKEN=${{secrets.JINAHUB_TOKEN}} jina hub push --force $exec_name --secret ${{secrets.ONNX_EXEC_SECRET}} server ${{env.GPU_TAG_ALIAS}}
 
       - name: Push TensorRT Executor
         id: push_tensorrt_executor