Unverified Commit 8da360aa authored by Linoy Tsaban's avatar Linoy Tsaban Committed by GitHub
Browse files

[training scripts] add tags of diffusers-training (#7206)

* add tags for diffusers training

* add tags for diffusers training

* add tags for diffusers training

* add tags for diffusers training

* add tags for diffusers training

* add tags for diffusers training

* add dora tags for drambooth lora scripts

* style
parent 869bad3e
...@@ -103,7 +103,14 @@ Special VAE used for training: {vae_path}. ...@@ -103,7 +103,14 @@ Special VAE used for training: {vae_path}.
inference=True, inference=True,
) )
tags = ["stable-diffusion-xl", "stable-diffusion-xl-diffusers", "text-to-image", "diffusers", "lora"] tags = [
"stable-diffusion-xl",
"stable-diffusion-xl-diffusers",
"text-to-image",
"diffusers",
"diffusers-training",
"lora",
]
model_card = populate_model_card(model_card, tags=tags) model_card = populate_model_card(model_card, tags=tags)
model_card.save(os.path.join(repo_folder, "README.md")) model_card.save(os.path.join(repo_folder, "README.md"))
......
...@@ -101,6 +101,7 @@ Special VAE used for training: {vae_path}. ...@@ -101,6 +101,7 @@ Special VAE used for training: {vae_path}.
"stable-diffusion-xl", "stable-diffusion-xl",
"stable-diffusion-xl-diffusers", "stable-diffusion-xl-diffusers",
"text-to-image", "text-to-image",
"diffusers-training",
"diffusers", "diffusers",
] ]
model_card = populate_model_card(model_card, tags=tags) model_card = populate_model_card(model_card, tags=tags)
......
...@@ -105,7 +105,14 @@ These are textual inversion adaption weights for {base_model}. You can find some ...@@ -105,7 +105,14 @@ These are textual inversion adaption weights for {base_model}. You can find some
inference=True, inference=True,
) )
tags = ["stable-diffusion", "stable-diffusion-diffusers", "text-to-image", "diffusers", "textual_inversion"] tags = [
"stable-diffusion",
"stable-diffusion-diffusers",
"text-to-image",
"diffusers",
"textual_inversion",
"diffusers-training",
]
model_card = populate_model_card(model_card, tags=tags) model_card = populate_model_card(model_card, tags=tags)
model_card.save(os.path.join(repo_folder, "README.md")) model_card.save(os.path.join(repo_folder, "README.md"))
......
...@@ -106,6 +106,7 @@ These are textual inversion adaption weights for {base_model}. You can find some ...@@ -106,6 +106,7 @@ These are textual inversion adaption weights for {base_model}. You can find some
"stable-diffusion-xl-diffusers", "stable-diffusion-xl-diffusers",
"text-to-image", "text-to-image",
"diffusers", "diffusers",
"diffusers-training",
"textual_inversion", "textual_inversion",
] ]
......
...@@ -81,6 +81,7 @@ tags: ...@@ -81,6 +81,7 @@ tags:
- wuerstchen - wuerstchen
- text-to-image - text-to-image
- diffusers - diffusers
- diffusers-training
- lora - lora
inference: true inference: true
--- ---
......
...@@ -82,6 +82,7 @@ tags: ...@@ -82,6 +82,7 @@ tags:
- wuerstchen - wuerstchen
- text-to-image - text-to-image
- diffusers - diffusers
- diffusers-training
inference: true inference: true
--- ---
""" """
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment