Fixup pipeline mlir -> vmfb

nod-ai · Mar 6, 2024 · c47605b · c47605b
1 parent 9ad5969
commit c47605b
Showing 1 changed file with 3 additions and 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
@@ -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