fix: keyword argument mismatch (#6895)

bamps53 · web-flow · commit 4a3d52850b2a · 2024-02-08T09:37:56.000+05:30
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,
     )

Original file line number	Diff line number	Diff line change
`@@ -92,7 +92,7 @@ def save_model_card(repo_id: str, images=None, base_model=str, prompt=str, repo_`
`92`	`92`	`from_training=True,`
`93`	`93`	`license="creativeml-openrail-m",`
`94`	`94`	`base_model=base_model,`
`95`		`- instance_prompt=prompt,`
	`95`	`+ prompt=prompt,`
`96`	`96`	`model_description=model_description,`
`97`	`97`	`inference=True,`
`98`	`98`	`)`
Original file line number	Diff line number	Diff line change
`@@ -97,7 +97,7 @@ def save_model_card(`
`97`	`97`	`from_training=True,`
`98`	`98`	`license="creativeml-openrail-m",`
`99`	`99`	`base_model=base_model,`
`100`		`- instance_prompt=prompt,`
	`100`	`+ prompt=prompt,`
`101`	`101`	`model_description=model_description,`
`102`	`102`	`inference=True,`
`103`	`103`	`)`
Original file line number	Diff line number	Diff line change
`@@ -99,7 +99,7 @@ def save_model_card(`
`99`	`99`	`from_training=True,`
`100`	`100`	`license="creativeml-openrail-m",`
`101`	`101`	`base_model=base_model,`
`102`		`- instance_prompt=prompt,`
	`102`	`+ prompt=prompt,`
`103`	`103`	`model_description=model_description,`
`104`	`104`	`inference=True,`
`105`	`105`	`)`
Original file line number	Diff line number	Diff line change
`@@ -122,7 +122,7 @@ def save_model_card(`
`122`	`122`	`from_training=True,`
`123`	`123`	`license="openrail++",`
`124`	`124`	`base_model=base_model,`
`125`		`- instance_prompt=instance_prompt,`
	`125`	`+ prompt=instance_prompt,`
`126`	`126`	`model_description=model_description,`
`127`	`127`	`widget=widget_dict,`
`128`	`128`	`)`