Fix for passing a path as attn_spec.

Author: eagarvey-amd

models/turbine_models/custom_models/sd_inference/sd_pipeline.py

@@ -244,7 +244,7 @@ def __init__(
             "exit_on_vmfb": False,
             "pipeline_dir": pipeline_dir,
             "input_mlir": None,
-            "attn_spec": None,
+            "attn_spec": attn_spec,
             "external_weights": None,
             "external_weight_path": None,
         }

models/turbine_models/custom_models/sdxl_inference/unet.py

@@ -182,12 +182,13 @@ def export_unet_model(
         submodel_name = "punet"
     else:
         submodel_name = "unet"
-    if (not decomp_attn) and use_punet:
-        attn_spec = "punet"
-    elif (not decomp_attn) and "gfx9" in target:
-        attn_spec = "mfma"
-    elif (not decomp_attn) and "gfx11" in target:
-        attn_spec = "wmma"
+    if not attn_spec:
+        if (not decomp_attn) and use_punet:
+            attn_spec = "punet"
+        elif (not decomp_attn) and "gfx9" in target:
+            attn_spec = "mfma"
+        elif (not decomp_attn) and "gfx11" in target:
+            attn_spec = "wmma"
     safe_name = utils.create_safe_name(
         hf_model_name,
         f"_bs{batch_size}_{max_length}_{height}x{width}_{precision}_{submodel_name}",