[WEB] Update to latest models.

Signed-Off-by: Gaurav Shukla <[email protected]>

Author: Shukla-Gaurav

web/models/stable_diffusion/opt_params.py

@@ -20,8 +20,8 @@ def get_unet(args):
             model_name = "unet_22nov_fp16_tuned"
             return get_shark_model(args, bucket, model_name, iree_flags)
         else:
-            bucket = "gs://shark_tank/prashant_nod"
-            model_name = "unet_23nov_fp16"
+            bucket = "gs://shark_tank/stable_diffusion"
+            model_name = "unet_1dec_fp16"
             iree_flags += [
                 "--iree-flow-enable-padding-linalg-ops",
                 "--iree-flow-linalg-ops-padding-size=32",
@@ -33,8 +33,8 @@ def get_unet(args):
 
     # Tuned model is not present for `fp32` case.
     if args.precision == "fp32":
-        bucket = "gs://shark_tank/prashant_nod"
-        model_name = "unet_23nov_fp32"
+        bucket = "gs://shark_tank/stable_diffusion"
+        model_name = "unet_1dec_fp32"
         iree_flags += [
             "--iree-flow-enable-conv-nchw-to-nhwc-transform",
             "--iree-flow-enable-padding-linalg-ops",
@@ -52,8 +52,8 @@ def get_vae(args):
             f"-iree-vulkan-target-triple={args.iree_vulkan_target_triple}"
         )
     if args.precision == "fp16":
-        bucket = "gs://shark_tank/prashant_nod"
-        model_name = "vae_22nov_fp16"
+        bucket = "gs://shark_tank/stable_diffusion"
+        model_name = "vae_1dec_fp16"
         iree_flags += [
             "--iree-flow-enable-conv-nchw-to-nhwc-transform",
             "--iree-flow-enable-padding-linalg-ops",
@@ -64,8 +64,8 @@ def get_vae(args):
         return get_shark_model(args, bucket, model_name, iree_flags)
 
     if args.precision == "fp32":
-        bucket = "gs://shark_tank/prashant_nod"
-        model_name = "vae_22nov_fp32"
+        bucket = "gs://shark_tank/stable_diffusion"
+        model_name = "vae_1dec_fp32"
         iree_flags += [
             "--iree-flow-enable-conv-nchw-to-nhwc-transform",
             "--iree-flow-enable-padding-linalg-ops",
@@ -82,8 +82,8 @@ def get_clip(args):
         iree_flags.append(
             f"-iree-vulkan-target-triple={args.iree_vulkan_target_triple}"
         )
-    bucket = "gs://shark_tank/prashant_nod"
-    model_name = "clip_18nov_fp32"
+    bucket = "gs://shark_tank/stable_diffusion"
+    model_name = "clip_1dec_fp32"
     iree_flags += [
         "--iree-flow-linalg-ops-padding-size=16",
         "--iree-flow-enable-padding-linalg-ops",