Update unet script for compile function signature change

eagarvey-amd · eagarvey-amd · commit dec8bc1ffcda · 2024-07-15T18:04:35.000-05:00
diff --git a/models/turbine_models/custom_models/sdxl_inference/unet.py b/models/turbine_models/custom_models/sdxl_inference/unet.py
@@ -208,7 +208,7 @@ def export_unet_model(
             mlir_source="file",
             return_path=not exit_on_vmfb,
             attn_spec=attn_spec,
-            flagset_keyword=["punet"] if use_punet else [],
+            flagset_keywords=["punet"] if use_punet else [],
         )
         return vmfb_path
     elif use_punet:

Original file line number	Diff line number	Diff line change
`@@ -208,7 +208,7 @@ def export_unet_model(`
`208`	`208`	`mlir_source="file",`
`209`	`209`	`return_path=not exit_on_vmfb,`
`210`	`210`	`attn_spec=attn_spec,`
`211`		`- flagset_keyword=["punet"] if use_punet else [],`
	`211`	`+ flagset_keywords=["punet"] if use_punet else [],`
`212`	`212`	`)`
`213`	`213`	`return vmfb_path`
`214`	`214`	`elif use_punet:`