more err handling for incompatible interps

n00mkrad · Dec 14, 2020 · cd9c5a3 · cd9c5a3
1 parent db32484
commit cd9c5a3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Code/OS/ESRGAN.cs b/Code/OS/ESRGAN.cs
@@ -273,7 +273,7 @@ private static void OutputHandler(object sendingProcess, DataReceivedEventArgs o
             if (data.Contains("UnpicklingError"))
                 Program.ShowMessage("Failed to load model!", "Error");
 
-            if (PreviewUI.currentMode == PreviewUI.Mode.Interp && data.Contains("must match the size of tensor b"))
+            if (PreviewUI.currentMode == PreviewUI.Mode.Interp && (data.Contains("must match the size of tensor b") || data.Contains("KeyError: 'model.")))
                 Program.ShowMessage("It seems like you tried to interpolate incompatible models!", "Error");
         }