From 7698be1bfb60f523e767ea5b42ee513c23d570ca Mon Sep 17 00:00:00 2001
From: Hawk Lee <l.y@live.cn>
Date: Wed, 21 Jan 2026 13:14:29 +0800
Subject: [PATCH] fix: Check type of ditto_config before unpacking to handle
 both dict and object

---
 aiia_ditto_nodes.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/aiia_ditto_nodes.py b/aiia_ditto_nodes.py
index 869f849..63ae440 100644
--- a/aiia_ditto_nodes.py
+++ b/aiia_ditto_nodes.py
@@ -652,6 +652,11 @@ class AIIA_DittoSampler:
         }
         
         
+        if isinstance(ditto_config, dict):
+            ditto_config_kwargs = ditto_config
+        else:
+            ditto_config_kwargs = vars(ditto_config)
+
         # Calling setup() creates fresh threads and queues
         # Wrap setup() with RestoreLogging to catch any init-time hijacking (e.g. MediaPipe/absl)
         # and restore it IMMEDIATELY before we start the long-running inference.
@@ -672,7 +677,7 @@ class AIIA_DittoSampler:
                 overall_ctrl_info=overall_ctrl_info,
                 ctrl_info=ctrl_info,
                 vad_timeline=dataset_alpha,
-                **vars(ditto_config)
+                **ditto_config_kwargs
             )
             
         # 4. Trigger Audio Feat Extraction & Pipeline