From 94a709d813faaa86bbdadbce1bd3b0888d89fe1e Mon Sep 17 00:00:00 2001
From: Rattus <rattus128@gmail.com>
Date: Tue, 13 Jan 2026 19:40:52 +1000
Subject: [PATCH] misc cleanup

---
 comfy/pinned_memory.py | 12 ++++--------
 execution.py           |  1 -
 requirements.txt       |  2 +-
 3 files changed, 5 insertions(+), 10 deletions(-)

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