Add lora tag to the model tags (#2103)

apolinario · patrickvonplaten · web-flow · commit 263b96804139 · 2023-01-25T12:17:59.000+01:00
* Add `lora` tag to the model tags

For lora training

* uP

Co-authored-by: Patrick von Platen &lt;patrick.v.platen@gmail.com&gt;
diff --git a/examples/dreambooth/train_dreambooth_lora.py b/examples/dreambooth/train_dreambooth_lora.py
@@ -74,6 +74,7 @@ def save_model_card(repo_name, images=None, base_model=str, prompt=str, repo_fol
 - stable-diffusion-diffusers
 - text-to-image
 - diffusers
+- lora
 inference: true
 ---
     """
diff --git a/examples/text_to_image/train_text_to_image_lora.py b/examples/text_to_image/train_text_to_image_lora.py
@@ -67,6 +67,7 @@ def save_model_card(repo_name, images=None, base_model=str, dataset_name=str, re
 - stable-diffusion-diffusers
 - text-to-image
 - diffusers
+- lora
 inference: true
 ---
     """
