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

support reference image

Browse Source
This commit is contained in:
Artiprocher
2025-04-11 16:23:14 +08:00
parent a3b4f235a0
commit 7a06a58f49
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
train_flux_reference.py
View File

@@ -212,6 +212,7 @@ if __name__ == '__main__':
), ),
], ],
dataset_weight=(4, 2, 2, 1), dataset_weight=(4, 2, 2, 1),
steps_per_epoch=args.steps_per_epoch,
) )
train_loader = torch.utils.data.DataLoader( train_loader = torch.utils.data.DataLoader(
dataset, dataset,