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

Update train_wan_t2v.py

Browse Source 
Included .jpeg extension for image type detection, preventing an error trying to the read image as a video format
This commit is contained in:
boopage
2025-03-05 11:13:11 +01:00
committed by GitHub
parent 41e3e4e157
commit 34a5ef8c15
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/wanvideo/train_wan_t2v.py
View File

@@ -70,7 +70,7 @@ class TextVideoDataset(torch.utils.data.Dataset):
def is_image(self, file_path):
file_ext_name = file_path.split(".")[-1]
if file_ext_name.lower() in ["jpg", "png", "webp"]:
if file_ext_name.lower() in ["jpg", "jpeg", "png", "webp"]:
return True
return False