fix: Add DITTO_AVAILABLE check to load_model to handle import failures gracefully

2026-01-21 21:23:42 +08:00
parent a35a069261
commit b1df5fcd5b
1 changed files with 3 additions and 0 deletions
@@ -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