If disable_small_model is set to false, make sure the flag is set for…

… run_deepvariant. PiperOrigin-RevId: 695370255
google · Nov 11, 2024 · 57e45ff · 57e45ff
1 parent 39cf496
commit 57e45ff
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/scripts/inference_deepvariant.sh b/scripts/inference_deepvariant.sh
@@ -694,7 +694,10 @@ function setup_args() {
   fi
   if [[ "${DISABLE_SMALL_MODEL}" = true ]]; then
     echo "Disabling small model"
-    extra_args+=( --disable_small_model "true")
+    extra_args+=( --disable_small_model )
+  else
+    echo "Enabling small model"
+    extra_args+=( --nodisable_small_model )
   fi
   # If you're running an older version (before 1.2) that doesn't have this flag,
   # you'll need to comment out this line.