Merge branch 'inference' into prep_model_weights

flexflow · Jul 21, 2023 · 60a6390 · 60a6390
2 parents b771046 + d62f193
commit 60a6390
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docker/run.sh b/docker/run.sh
@@ -58,8 +58,8 @@ else
   cuda_version_hyphen=""
 fi
 
-# Check that image exists
-if docker image inspect "${image}-${FF_GPU_BACKEND}${cuda_version_hyphen}":latest > /dev/null || true ; then
+# Check that image exists, if fails, print the default error message.
+if [[ "$(docker images -q "$image"-"$FF_GPU_BACKEND""$cuda_version_hyphen":latest 2> /dev/null)" == "" ]]; then
   echo ""
   echo "To download the docker image, run:"
   echo "    FF_GPU_BACKEND=${FF_GPU_BACKEND} cuda_version=${cuda_version} $(pwd)/pull.sh $image"