From 1f81fbe274e67c843283e69eb8f00bb56f75ffc4 Mon Sep 17 00:00:00 2001 From: Hammond Liu Date: Tue, 25 Jun 2024 10:01:20 +0800 Subject: [PATCH] Fix redundant pipe init in sd3 lora (#8680) Co-authored-by: Sayak Paul --- examples/dreambooth/train_dreambooth_lora_sd3.py | 7 ------- 1 file changed, 7 deletions(-) diff --git a/examples/dreambooth/train_dreambooth_lora_sd3.py b/examples/dreambooth/train_dreambooth_lora_sd3.py index c8ee825ccc..530634fd68 100644 --- a/examples/dreambooth/train_dreambooth_lora_sd3.py +++ b/examples/dreambooth/train_dreambooth_lora_sd3.py @@ -1638,13 +1638,6 @@ def main(args): save_directory=args.output_dir, transformer_lora_layers=transformer_lora_layers ) - 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(