(SD) Fix precompiled vmfb loading for schedulers. (#771)

monorimet · web-flow · commit eb61c143d293 · 2024-07-12T11:47:44.000-07:00
Fixes a filename constructor for expected vmfb filename so that it
matches what is exported.
diff --git a/models/turbine_models/custom_models/sd_inference/sd_pipeline.py b/models/turbine_models/custom_models/sd_inference/sd_pipeline.py
@@ -408,7 +408,7 @@ def load_scheduler(
             )
             scheduler_path = os.path.join(
                 self.pipeline_dir,
-                utils.create_safe_name(self.base_model_name, scheduler_uid),
+                utils.create_safe_name(self.base_model_name, scheduler_uid) + ".vmfb",
             )
             if not os.path.exists(scheduler_path):
                 self.export_submodel("scheduler")

Original file line number	Diff line number	Diff line change
`@@ -408,7 +408,7 @@ def load_scheduler(`
`408`	`408`	`)`
`409`	`409`	`scheduler_path = os.path.join(`
`410`	`410`	`self.pipeline_dir,`
`411`		`- utils.create_safe_name(self.base_model_name, scheduler_uid),`
	`411`	`+ utils.create_safe_name(self.base_model_name, scheduler_uid) + ".vmfb",`
`412`	`412`	`)`
`413`	`413`	`if not os.path.exists(scheduler_path):`
`414`	`414`	`self.export_submodel("scheduler")`