Update opt_params.py

2026-01-13 07:48:01 -05:00 · 2022-12-19 23:39:04 +05:30
1 changed files with 1 additions and 4 deletions
--- a/shark/examples/shark_inference/stable_diffusion/opt_params.py
+++ b/shark/examples/shark_inference/stable_diffusion/opt_params.py
@@ -114,10 +114,7 @@ def get_vae():
            bucket = "gs://shark_tank/stable_diffusion"
            model_name = "vae_19dec_fp16"
            if args.version == "v2.1base":
-                if args.max_length == 64:
-                    model_name = "vae_19dec_v2p1base_fp16_64"
-                else:
-                    model_name = "vae2base_19dec_fp16"
+                model_name = "vae2base_19dec_fp16"
            if args.version == "v2.1":
                model_name = "vae2_19dec_fp16"
            iree_flags += [