diff --git a/configs/train_i2v_dmd.yaml b/configs/train_i2v_dmd.yaml
index e7df4d71..7d974473 100644
--- a/configs/train_i2v_dmd.yaml
+++ b/configs/train_i2v_dmd.yaml
@@ -77,6 +77,7 @@ data:
 inference:
   sampling_steps: 4
   sink_size: 0
+  multi_shot_sink: true
   multi_shot_rope_offset: 8
   independent_first_frame: true