Fix model fusion

IAHispano · Feb 28, 2024 · ef06b62 · ef06b62
1 parent c65dfef
commit ef06b62
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/core.py b/core.py
@@ -394,7 +394,9 @@ def run_model_information_script(pth_path):
 
 # Model blender
 def run_model_blender_script(model_name, pth_path_1, pth_path_2, ratio):
-    model_blender(model_name, pth_path_1, pth_path_2, ratio)
+    message, model_blended = model_blender(model_name, pth_path_1, pth_path_2, ratio)
+    return message, model_blended
+
 
 
 # Tensorboard