From da113364dff2f52cc26690d617d766272ed643ca Mon Sep 17 00:00:00 2001
From: hysts <hysts@users.noreply.github.com>
Date: Tue, 31 Jan 2023 16:14:25 +0900
Subject: [PATCH] Add instance prompt to model card of lora dreambooth example
 (#2112)

---
 examples/dreambooth/train_dreambooth_lora.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/examples/dreambooth/train_dreambooth_lora.py b/examples/dreambooth/train_dreambooth_lora.py
index 86ac3d7e77..8bc4e2da41 100644
--- 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