fix: keyword argument mismatch (huggingface#6895)

AmericanPresidentJimmyCarter · Apr 26, 2024 · c85ea2b · c85ea2b
1 parent e0fa92f
commit c85ea2b
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/examples/custom_diffusion/train_custom_diffusion.py b/examples/custom_diffusion/train_custom_diffusion.py
@@ -92,7 +92,7 @@ def save_model_card(repo_id: str, images=None, base_model=str, prompt=str, repo_
         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
@@ -97,7 +97,7 @@ def save_model_card(
         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
@@ -99,7 +99,7 @@ def save_model_card(
         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
@@ -122,7 +122,7 @@ def save_model_card(
         from_training=True,
         license="openrail++",
         base_model=base_model,
-        instance_prompt=instance_prompt,
+        prompt=instance_prompt,
         model_description=model_description,
         widget=widget_dict,
     )