add tensorflow-probability to resource tar

alibaba · Nov 13, 2024 · 6cd69b9 · 6cd69b9
1 parent 1f507d3
commit 6cd69b9
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/pai_jobs/deploy_ext.sh b/pai_jobs/deploy_ext.sh
@@ -145,18 +145,23 @@ fi
 
 if [ ! -d "tensorflow_probability" ]
 then
+  if [ $is_tf15 -gt 0 ]; then
+    tfp_version='0.8.0'
+  else
+    tfp_version='0.5.0'
+  fi
   if [ ! -e "tensorflow_probability" ]
   then
-    wget http://easyrec.oss-cn-beijing.aliyuncs.com/3rdparty/probability-0.5.0.tar.gz
+    wget http://easyrec.oss-cn-beijing.aliyuncs.com/3rdparty/probability-${tfp_version}.tar.gz
     if [ $? -ne 0 ]
     then
       echo "tensorflow_probability download failed."
     fi
   fi
-  tar -xzvf probability-0.5.0.tar.gz --strip-components=1 probability-0.5.0/tensorflow_probability
+  tar -xzvf probability-${tfp_version}.tar.gz --strip-components=1 probability-${tfp_version}/tensorflow_probability
   rm -rf tensorflow_probability/examples
   rm -rf tensorflow_probability/g3doc
-  rm -rf probability-0.5.0.tar.gz
+  rm -rf probability-${tfp_version}.tar.gz
 fi
 
 tar -cvzhf $RES_PATH easy_rec datahub lz4 cprotobuf kafka faiss tensorflow_probability run.py