Skip to content

Commit

Permalink
Update unet script for compile function signature change
Browse files Browse the repository at this point in the history
  • Loading branch information
eagarvey-amd committed Jul 15, 2024
1 parent 3e1b60f commit dec8bc1
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion models/turbine_models/custom_models/sdxl_inference/unet.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit dec8bc1

Please sign in to comment.