From 8dfdeadc4718a167d50a9eea40f11c2abfe2fad5 Mon Sep 17 00:00:00 2001
From: Hawk Lee <l.y@live.cn>
Date: Wed, 21 Jan 2026 13:58:20 +0800
Subject: [PATCH] fix: Pass total_frames to SDK setup to enable Ditto console
 progress bar

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

diff --git a/aiia_ditto_nodes.py b/aiia_ditto_nodes.py
index 054337c..9a603ed 100644
--- a/aiia_ditto_nodes.py
+++ b/aiia_ditto_nodes.py
@@ -673,6 +673,7 @@ class AIIA_DittoSampler:
                 source_path=None, 
                 emo=emo_idx,
                 source_image_pil=ref_image_pil,
+                total_frames=num_frames, # Make sure to pass this for progress bar!
                 output_path=None,
                 N_d=num_frames,
                 sampling_timesteps=sampling_steps,