Avoid exceptions to be silenced

4a66d2fb · Kohaku-Blueleaf · 81126027 · 4a66d2fb
Commit 4a66d2fb authored Jan 20, 2024 by Kohaku-Blueleaf
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 6 deletions

modules/devices.py modules/devices.py +5 -6

No files found.
--- a/modules/devices.py
+++ b/modules/devices.py
@@ -178,12 +178,11 @@ def manual_cast(target_dtype):
    try:
        yield None
    finally:
-        if not applied:
-            return
-        for module_type in patch_module_list:
-            if hasattr(module_type, "org_forward"):
-                module_type.forward = module_type.org_forward
-                delattr(module_type, "org_forward")
+        if applied:
+            for module_type in patch_module_list:
+                if hasattr(module_type, "org_forward"):
+                    module_type.forward = module_type.org_forward
+                    delattr(module_type, "org_forward")


 def autocast(disable=False):