From c47605b66fed7328e74bdcd73e6da2f27f032c7d Mon Sep 17 00:00:00 2001 From: Ean Garvey Date: Wed, 6 Mar 2024 16:33:31 -0600 Subject: [PATCH] Fixup pipeline mlir -> vmfb --- .../custom_models/sdxl_inference/sdxl_pipeline.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/models/turbine_models/custom_models/sdxl_inference/sdxl_pipeline.py b/models/turbine_models/custom_models/sdxl_inference/sdxl_pipeline.py index 7dc09b08c..af2677075 100644 --- a/models/turbine_models/custom_models/sdxl_inference/sdxl_pipeline.py +++ b/models/turbine_models/custom_models/sdxl_inference/sdxl_pipeline.py @@ -182,7 +182,9 @@ def export_submodel(args, submodel): return clip_2_vmfb, clip_external_weight_path case "pipeline": pipeline_file = ( - "sdxl_sched_unet_bench_" + "f32" if args.precision == "fp32" else "f16" + "sdxl_sched_unet_bench_" + "f32" + if args.precision == "fp32" + else "sdxl_sched_unet_bench" + "f16" ) pipeline_vmfb = utils.compile_to_vmfb( os.path.join( @@ -196,7 +198,6 @@ def export_submodel(args, submodel): const_expr_hoisting=False, mlir_source="file", ) - breakpoint() return pipeline_vmfb, None