Fix a bug in SD vmfb naming

nod-ai · Sep 17, 2024 · 1d47593 · 1d47593
1 parent 5eb013d
commit 1d47593
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/models/turbine_models/custom_models/sd_inference/utils.py b/models/turbine_models/custom_models/sd_inference/utils.py
@@ -299,7 +299,8 @@ def compile_to_vmfb(
         safe_mlir_name = safe_name
     else:
         safe_vmfb_name = safe_name
-        safe_mlir_name = "".join(safe_name.split(target_triple))
+        (dir, file) = os.path.split(safe_name)
+        safe_mlir_name = os.path.join(dir, "".join(file.split(target_triple)))
 
     if mlir_source == "file":
         flatbuffer_blob = ireec.compile_file(