diff --git a/comfy/model_management.py b/comfy/model_management.py
index bcf1399c4..3b39d6080 100644
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@@ -1801,7 +1801,7 @@ def debug_memory_summary():
         return torch.cuda.memory.memory_summary()
     return ""
 
-class InterruptProcessingException(Exception):
+class InterruptProcessingException(BaseException):
     pass
 
 interrupt_processing_mutex = threading.RLock()