fix videodataset to load images

2026-03-18 22:08:13 +00:00 · 2025-06-24 11:38:43 +08:00
parent 93194f44e8
commit a37a83ecc3
1 changed files with 2 additions and 1 deletions
--- 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):