From 2ad5af11200a43c25f594e1ddfd28ddda6ab555b Mon Sep 17 00:00:00 2001
From: John Pollock <pollockjj@gmail.com>
Date: Fri, 8 May 2026 18:45:08 +0000
Subject: [PATCH] Refresh aimdo readiness after init

---
 __init__.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/__init__.py b/__init__.py
index d043552..1ad2674 100644
--- a/__init__.py
+++ b/__init__.py
@@ -480,8 +480,10 @@ def _initialize_aimdo_visible_cuda_devices():
         logger.info(f"[MultiGPU] Initializing comfy_aimdo for CUDA device {device_index}")
         initialized = bool(init_device(device_index))
         logger.info(f"[MultiGPU] comfy_aimdo init_device({device_index}) -> {initialized}")
+        _aimdo_readiness_cache[device_index] = initialized
         if initialized:
             _aimdo_initialized_devices.add(device_index)
+            _aimdo_legacy_fallback_devices.discard(device_index)
             initialized_any = True
 
     return initialized_any