Potentially redundant code in `train_dreambooth_lora_sd3.py`

[hmdliu]

Describe the bug

The sd3 pipeline has been initialized twice in the example code for train_dreambooth_lora_sd3.py.

This is the code from L1641-L1655. You may also check it via this link.

 pipeline = StableDiffusion3Pipeline.from_pretrained(
    args.pretrained_model_name_or_path,
    revision=args.revision,
    variant=args.variant,
    torch_dtype=weight_dtype,
)

# Final inference
# Load previous pipeline
pipeline = StableDiffusion3Pipeline.from_pretrained(
    args.pretrained_model_name_or_path,
    revision=args.revision,
    variant=args.variant,
    torch_dtype=weight_dtype,
)

Reproduction

n/a

Logs

No response

System Info

n/a

Who can help?

@sayakpaul

[sayakpaul]

Yes, you are right. Would you mind opening a PR to fix this?

[hmdliu]

Sure! Thanks for the reply. The PR has been opened: link.