diff --git a/comfy/pinned_memory.py b/comfy/pinned_memory.py index 923872dac..dfb5fcfcd 100644 --- a/comfy/pinned_memory.py +++ b/comfy/pinned_memory.py @@ -14,14 +14,10 @@ def pin_memory(module): #FIXME: This is a RAM cache trigger event params = [ module.weight, module.bias ] size = comfy.memory_management.vram_aligned_size(params) - try: - pin = torch.empty((size,), dtype=torch.uint8) - if comfy.model_management.pin_memory(pin): - module._pin = pin - else: - module.pin_failed = True - return False - except: + pin = torch.empty((size,), dtype=torch.uint8) + if comfy.model_management.pin_memory(pin): + module._pin = pin + else: module.pin_failed = True return False return True diff --git a/execution.py b/execution.py index fe162db26..197e13f38 100644 --- a/execution.py +++ b/execution.py @@ -14,7 +14,6 @@ from contextlib import nullcontext import torch -import comfy.pinned_memory import comfy.memory_management import comfy.model_management from latent_preview import set_preview_method diff --git a/requirements.txt b/requirements.txt index 950f51c15..2574f415b 100644 --- a/requirements.txt +++ b/requirements.txt @@ -22,7 +22,7 @@ alembic SQLAlchemy av>=14.2.0 comfy-kitchen>=0.2.6 -comfy-aimdo>=0.1.0 +comfy-aimdo>=0.1.1 #non essential dependencies: kornia>=0.7.1