From b1df5fcd5bcb8ade7886de3e4171f8aeea76185b Mon Sep 17 00:00:00 2001
From: Hawk Lee <l.y@live.cn>
Date: Wed, 21 Jan 2026 21:23:42 +0800
Subject: [PATCH] fix: Add DITTO_AVAILABLE check to load_model to handle import
 failures gracefully

---
 aiia_ditto_nodes.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/aiia_ditto_nodes.py b/aiia_ditto_nodes.py
index 36522a1..0473da0 100644
--- a/aiia_ditto_nodes.py
+++ b/aiia_ditto_nodes.py
@@ -86,6 +86,9 @@ class AIIA_DittoLoader:
     CATEGORY = "AIIA/Ditto"
 
     def load_model(self, model_name, device):
+        if not DITTO_AVAILABLE:
+            raise RuntimeError("Ditto libraries failed to import. Please check console logs for details (missing dependencies?).")
+            
         print(f"[AIIA_DittoLoader] Loading Ditto model: {model_name} on {device}")
         
         # 1. Prepare Model Paths