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

bugfix

Browse Source
This commit is contained in:
Artiprocher
2025-03-18 19:36:58 +08:00
parent a8ce9fef33
commit ce848a3d1a
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
examples/wanvideo/train_wan_t2v.py
View File

@@ -546,7 +546,8 @@ def data_process(args):
num_frames=args.num_frames, num_frames=args.num_frames,
height=args.height, height=args.height,
width=args.width, width=args.width,
is_i2v=args.image_encoder_path is not None is_i2v=args.image_encoder_path is not None,
target_fps=args.target_fps,
) )
dataloader = torch.utils.data.DataLoader( dataloader = torch.utils.data.DataLoader(
dataset, dataset,