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

Merge pull request #412 from boopage/patch-1

Browse Source 
Update train_wan_t2v.py - include .jpeg for image detection
This commit is contained in:
Zhongjie Duan
2025-03-06 12:46:43 +08:00
committed by GitHub
parent 41e3e4e157 34a5ef8c15
commit a58c193d0c
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