add clip test poc

nod-ai · Jun 19, 2024 · fd1543b · fd1543b
1 parent 92be65b
commit fd1543b
Show file tree

Hide file tree

Showing 3 changed files with 559 additions and 2 deletions.
diff --git a/models/turbine_models/custom_models/sd3_inference/sd3_cmd_opts.py b/models/turbine_models/custom_models/sd3_inference/sd3_cmd_opts.py
@@ -406,7 +406,7 @@ def is_valid_file(arg):
 )
 
 p.add_argument(
-    "--unet_flags",
+    "--mmdit_flags",
     type=str,
     default="",
     help="extra iree-compile options to send for compiling unet. Only use this for testing bleeding edge flags! Any default options should be added to sd_inference/utils.py",

diff --git a/models/turbine_models/custom_models/sd3_inference/sd3_mmdit.py b/models/turbine_models/custom_models/sd3_inference/sd3_mmdit.py
@@ -301,7 +301,7 @@ class CompiledMmdit(CompiledModule):
         args.external_weight_path,
         args.device,
         args.iree_target_triple,
-        args.ireec_flags + args.attn_flags + args.unet_flags,
+        args.ireec_flags + args.attn_flags + args.mmdit_flags,
         args.decomp_attn,
         attn_spec=args.attn_spec,
         input_mlir=args.input_mlir,