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):