From 4a3d52850b2a1a3da47c91525b8899465b76606e Mon Sep 17 00:00:00 2001
From: camaro <37071446+bamps53@users.noreply.github.com>
Date: Thu, 8 Feb 2024 13:07:56 +0900
Subject: [PATCH] fix: keyword argument mismatch (#6895)

---
 examples/custom_diffusion/train_custom_diffusion.py | 2 +-
 examples/dreambooth/train_dreambooth.py             | 2 +-
 examples/dreambooth/train_dreambooth_lora.py        | 2 +-
 examples/dreambooth/train_dreambooth_lora_sdxl.py   | 2 +-
 4 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/examples/custom_diffusion/train_custom_diffusion.py b/examples/custom_diffusion/train_custom_diffusion.py
index 0ea2732a95..0e59db2114 100644
--- a/examples/custom_diffusion/train_custom_diffusion.py
+++ b/examples/custom_diffusion/train_custom_diffusion.py
@@ -92,7 +92,7 @@ These are Custom Diffusion adaption weights for {base_model}. The weights were t
         from_training=True,
         license="creativeml-openrail-m",
         base_model=base_model,
-        instance_prompt=prompt,
+        prompt=prompt,
         model_description=model_description,
         inference=True,
     )
diff --git a/examples/dreambooth/train_dreambooth.py b/examples/dreambooth/train_dreambooth.py
index 4d899ef56e..4847e214bc 100644
--- a/examples/dreambooth/train_dreambooth.py
+++ b/examples/dreambooth/train_dreambooth.py
@@ -97,7 +97,7 @@ DreamBooth for the text encoder was enabled: {train_text_encoder}.
         from_training=True,
         license="creativeml-openrail-m",
         base_model=base_model,
-        instance_prompt=prompt,
+        prompt=prompt,
         model_description=model_description,
         inference=True,
     )
diff --git a/examples/dreambooth/train_dreambooth_lora.py b/examples/dreambooth/train_dreambooth_lora.py
index f0c47821b0..5aa2bc16b8 100644
--- a/examples/dreambooth/train_dreambooth_lora.py
+++ b/examples/dreambooth/train_dreambooth_lora.py
@@ -99,7 +99,7 @@ LoRA for the text encoder was enabled: {train_text_encoder}.
         from_training=True,
         license="creativeml-openrail-m",
         base_model=base_model,
-        instance_prompt=prompt,
+        prompt=prompt,
         model_description=model_description,
         inference=True,
     )
diff --git a/examples/dreambooth/train_dreambooth_lora_sdxl.py b/examples/dreambooth/train_dreambooth_lora_sdxl.py
index 8df61f1325..46bf40413d 100644
--- a/examples/dreambooth/train_dreambooth_lora_sdxl.py
+++ b/examples/dreambooth/train_dreambooth_lora_sdxl.py
@@ -122,7 +122,7 @@ Weights for this model are available in Safetensors format.
         from_training=True,
         license="openrail++",
         base_model=base_model,
-        instance_prompt=instance_prompt,
+        prompt=instance_prompt,
         model_description=model_description,
         widget=widget_dict,
     )