From a37a83ecc33d93745a8e636728c0ee993a4b8562 Mon Sep 17 00:00:00 2001 From: Artiprocher Date: Tue, 24 Jun 2025 11:38:43 +0800 Subject: [PATCH] fix videodataset to load images --- diffsynth/trainers/utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/diffsynth/trainers/utils.py b/diffsynth/trainers/utils.py index d627dab..a15a1d2 100644 --- a/diffsynth/trainers/utils.py +++ b/diffsynth/trainers/utils.py @@ -134,7 +134,8 @@ class VideoDataset(torch.utils.data.Dataset): def load_image(self, file_path): image = Image.open(file_path).convert("RGB") image = self.crop_and_resize(image, *self.get_height_width(image)) - return image + frames = [image] + return frames def is_image(self, file_path):