Skip to content

Commit 4a3d528

Browse files
authored
fix: keyword argument mismatch (huggingface#6895)
1 parent 97d004b commit 4a3d528

File tree

4 files changed

+4
-4
lines changed

4 files changed

+4
-4
lines changed

examples/custom_diffusion/train_custom_diffusion.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ def save_model_card(repo_id: str, images=None, base_model=str, prompt=str, repo_
9292
from_training=True,
9393
license="creativeml-openrail-m",
9494
base_model=base_model,
95-
instance_prompt=prompt,
95+
prompt=prompt,
9696
model_description=model_description,
9797
inference=True,
9898
)

examples/dreambooth/train_dreambooth.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,7 @@ def save_model_card(
9797
from_training=True,
9898
license="creativeml-openrail-m",
9999
base_model=base_model,
100-
instance_prompt=prompt,
100+
prompt=prompt,
101101
model_description=model_description,
102102
inference=True,
103103
)

examples/dreambooth/train_dreambooth_lora.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,7 @@ def save_model_card(
9999
from_training=True,
100100
license="creativeml-openrail-m",
101101
base_model=base_model,
102-
instance_prompt=prompt,
102+
prompt=prompt,
103103
model_description=model_description,
104104
inference=True,
105105
)

examples/dreambooth/train_dreambooth_lora_sdxl.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -122,7 +122,7 @@ def save_model_card(
122122
from_training=True,
123123
license="openrail++",
124124
base_model=base_model,
125-
instance_prompt=instance_prompt,
125+
prompt=instance_prompt,
126126
model_description=model_description,
127127
widget=widget_dict,
128128
)

0 commit comments

Comments
 (0)