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

minor fix

Browse Source
This commit is contained in:
mi804
2025-08-06 10:58:53 +08:00
parent 4299c999b5
commit 3915bc3ee6
2 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
diffsynth/trainers/utils.py
View File

@@ -404,6 +404,7 @@ def launch_training_task(
) )
model, optimizer, dataloader, scheduler = accelerator.prepare(model, optimizer, dataloader, scheduler) model, optimizer, dataloader, scheduler = accelerator.prepare(model, optimizer, dataloader, scheduler)
global_steps = 0
for epoch_id in range(num_epochs): for epoch_id in range(num_epochs):
for step_id, data in enumerate(tqdm(dataloader)): for step_id, data in enumerate(tqdm(dataloader)):
with accelerator.accumulate(model): with accelerator.accumulate(model):

2
examples/qwen_image/model_training/lora/Qwen-Image.sh
View File

@@ -13,5 +13,5 @@ accelerate launch examples/qwen_image/model_training/train.py \
--lora_rank 32 \ --lora_rank 32 \
--align_to_opensource_format \ --align_to_opensource_format \
--use_gradient_checkpointing \ --use_gradient_checkpointing \
--num_workers 8 \ --dataset_num_workers 8 \
--find_unused_parameters --find_unused_parameters