mirror of
https://github.com/modelscope/DiffSynth-Studio.git
synced 2026-03-18 22:08:13 +00:00
minor fix
This commit is contained in:
@@ -404,6 +404,7 @@ def launch_training_task(
|
||||
)
|
||||
model, optimizer, dataloader, scheduler = accelerator.prepare(model, optimizer, dataloader, scheduler)
|
||||
|
||||
global_steps = 0
|
||||
for epoch_id in range(num_epochs):
|
||||
for step_id, data in enumerate(tqdm(dataloader)):
|
||||
with accelerator.accumulate(model):
|
||||
|
||||
@@ -13,5 +13,5 @@ accelerate launch examples/qwen_image/model_training/train.py \
|
||||
--lora_rank 32 \
|
||||
--align_to_opensource_format \
|
||||
--use_gradient_checkpointing \
|
||||
--num_workers 8 \
|
||||
--dataset_num_workers 8 \
|
||||
--find_unused_parameters
|
||||
|
||||
Reference in New Issue
Block a user