minor fix

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

View File

@@ -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):

View File

@@ -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