Spaces:
Running
on
Zero
Running
on
Zero
fix comfy dep error
Browse files- nodes/base_vibevoice.py +6 -3
nodes/base_vibevoice.py
CHANGED
|
@@ -863,9 +863,12 @@ class BaseVibeVoiceNode:
|
|
| 863 |
|
| 864 |
except Exception as e:
|
| 865 |
# Re-raise interruption exceptions without wrapping
|
| 866 |
-
|
| 867 |
-
|
| 868 |
-
|
|
|
|
|
|
|
|
|
|
| 869 |
|
| 870 |
# For real errors, log and re-raise with context
|
| 871 |
logger.error(f"VibeVoice generation failed: {e}")
|
|
|
|
| 863 |
|
| 864 |
except Exception as e:
|
| 865 |
# Re-raise interruption exceptions without wrapping
|
| 866 |
+
try:
|
| 867 |
+
import comfy.model_management as mm
|
| 868 |
+
if isinstance(e, mm.InterruptProcessingException):
|
| 869 |
+
raise # Let the interruption propagate
|
| 870 |
+
except ImportError:
|
| 871 |
+
pass
|
| 872 |
|
| 873 |
# For real errors, log and re-raise with context
|
| 874 |
logger.error(f"VibeVoice generation failed: {e}")
|