diff --git a/examples/custom_diffusion/train_custom_diffusion.py b/examples/custom_diffusion/train_custom_diffusion.py index 0ea2732a959f..0e59db211492 100644 --- 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 index 4d899ef56e37..4847e214bc48 100644 --- 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 index f0c47821b0c9..5aa2bc16b84e 100644 --- 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 index 8df61f132510..46bf40413d7c 100644 --- 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, )