Add instance prompt to model card of lora dreambooth example (#2112)

da113364 · hysts · GitHub · 44f6bc81 · da113364
Unverified Commit da113364 authored Jan 31, 2023 by hysts Committed by GitHub Jan 31, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

examples/dreambooth/train_dreambooth_lora.py examples/dreambooth/train_dreambooth_lora.py +1 -0

No files found.
--- a/examples/dreambooth/train_dreambooth_lora.py
+++ b/examples/dreambooth/train_dreambooth_lora.py
@@ -69,6 +69,7 @@ def save_model_card(repo_name, images=None, base_model=str, prompt=str, repo_fol
 ---
 license: creativeml-openrail-m
 base_model: {base_model}
+instance_prompt: {prompt}
 tags:
 - stable-diffusion
 - stable-diffusion-diffusers