From f6418004bb441598c83f5dd60520f1ae91a247d6 Mon Sep 17 00:00:00 2001
From: twu <twu@30>
Date: Fri, 22 Aug 2025 03:00:35 +0000
Subject: [PATCH] as numframe limit is impled in reader, add that

---
 diffsynth/trainers/utils.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/diffsynth/trainers/utils.py b/diffsynth/trainers/utils.py
index c4b5d92..c5e0f19 100644
--- a/diffsynth/trainers/utils.py
+++ b/diffsynth/trainers/utils.py
@@ -294,6 +294,13 @@ class VideoDataset(torch.utils.data.Dataset):
                         last_match = idx + last_match
                         break
             frames = _frames
+        num_frames = len(frames)
+        if num_frames > self.num_frames:
+            num_frames = self.num_frames
+        else:
+            while num_frames > 1 and num_frames % self.time_division_factor != self.time_division_remainder:
+                num_frames -= 1
+        frames = frames[:num_frames]
         return frames
     
     def load_video(self, file_path):