From bffdb901ed475dbb44cceed7a829f9a2c779494a Mon Sep 17 00:00:00 2001
From: Kared <101872541+huarzone@users.noreply.github.com>
Date: Sat, 13 Sep 2025 21:01:44 +0800
Subject: [PATCH] fix load gif

---
 diffsynth/trainers/unified_dataset.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/diffsynth/trainers/unified_dataset.py b/diffsynth/trainers/unified_dataset.py
index c764ebd..4e94ab8 100644
--- a/diffsynth/trainers/unified_dataset.py
+++ b/diffsynth/trainers/unified_dataset.py
@@ -269,7 +269,10 @@ class UnifiedDataset(torch.utils.data.Dataset):
         return RouteByType(operator_map=[
             (str, ToAbsolutePath(base_path) >> RouteByExtensionName(operator_map=[
                 (("jpg", "jpeg", "png", "webp"), LoadImage() >> ImageCropAndResize(height, width, max_pixels, height_division_factor, width_division_factor) >> ToList()),
-                (("gif",), LoadGIF(num_frames, time_division_factor, time_division_remainder) >> ImageCropAndResize(height, width, max_pixels, height_division_factor, width_division_factor)),
+                (("gif",), LoadGIF(
+                    num_frames, time_division_factor, time_division_remainder,
+                    frame_processor=ImageCropAndResize(height, width, max_pixels, height_division_factor, width_division_factor),
+                )),
                 (("mp4", "avi", "mov", "wmv", "mkv", "flv", "webm"), LoadVideo(
                     num_frames, time_division_factor, time_division_remainder,
                     frame_processor=ImageCropAndResize(height, width, max_pixels, height_division_factor, width_division_factor),