theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-24 10:18:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix videodataset to load images

Browse Source
This commit is contained in:
Artiprocher
2025-06-24 11:38:43 +08:00
parent 93194f44e8
commit a37a83ecc3
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
diffsynth/trainers/utils.py
View File

@@ -134,7 +134,8 @@ class VideoDataset(torch.utils.data.Dataset):
def load_image(self, file_path): def load_image(self, file_path):
image = Image.open(file_path).convert("RGB") image = Image.open(file_path).convert("RGB")
image = self.crop_and_resize(image, *self.get_height_width(image)) image = self.crop_and_resize(image, *self.get_height_width(image))
return image frames = [image]
return frames
def is_image(self, file_path): def is_image(self, file_path):