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

refine script

Browse Source
This commit is contained in:
Artiprocher
2024-06-05 13:34:55 +08:00
parent 78f53a0754
commit 0d03cb31ad
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/hunyuan_dit/train_hunyuan_dit_lora.py
View File

@@ -256,7 +256,7 @@ if __name__ == '__main__':
# dataset and data loader # dataset and data loader
dataset = TextImageDataset( dataset = TextImageDataset(
args.dataset_path, args.dataset_path,
steps_per_epoch=args.steps_per_epoch, steps_per_epoch=args.steps_per_epoch * args.batch_size,
height=args.height, height=args.height,
width=args.width, width=args.width,
center_crop=args.center_crop, center_crop=args.center_crop,