diff --git a/examples/dreambooth/train_dreambooth_lora_hidream.py b/examples/dreambooth/train_dreambooth_lora_hidream.py
index f368fb809e73..46a022733cab 100644
--- a/examples/dreambooth/train_dreambooth_lora_hidream.py
+++ b/examples/dreambooth/train_dreambooth_lora_hidream.py
@@ -1771,16 +1771,16 @@ def get_sigmas(timesteps, n_dim=4, dtype=torch.float32):
             free_memory()
 
         validation_prompt = args.validation_prompt if args.validation_prompt else args.final_validation_prompt
-        save_model_card(
-            (args.hub_model_id or Path(args.output_dir).name) if not args.push_to_hub else repo_id,
-            images=images,
-            base_model=args.pretrained_model_name_or_path,
-            instance_prompt=args.instance_prompt,
-            validation_prompt=validation_prompt,
-            repo_folder=args.output_dir,
-        )
 
         if args.push_to_hub:
+            save_model_card(
+                (args.hub_model_id or Path(args.output_dir).name) if not args.push_to_hub else repo_id,
+                images=images,
+                base_model=args.pretrained_model_name_or_path,
+                instance_prompt=args.instance_prompt,
+                validation_prompt=validation_prompt,
+                repo_folder=args.output_dir,
+            )
             upload_folder(
                 repo_id=repo_id,
                 folder_path=args.output_dir,
@@ -1789,6 +1789,7 @@ def get_sigmas(timesteps, n_dim=4, dtype=torch.float32):
             )
 
         images = None
+        del pipeline
 
     accelerator.end_training()