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

linoytsaban · web-flow · commit 8da360aa12d3 · 2024-03-04T22:17:25.000+05:30
* 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
diff --git a/examples/advanced_diffusion_training/train_dreambooth_lora_sd15_advanced.py b/examples/advanced_diffusion_training/train_dreambooth_lora_sd15_advanced.py
@@ -77,6 +77,7 @@
 
 def save_model_card(
     repo_id: str,
+    use_dora: bool,
     images=None,
     base_model=str,
     train_text_encoder=False,
@@ -88,6 +89,7 @@ def save_model_card(
     vae_path=None,
 ):
     img_str = "widget:\n"
+    lora = "lora" if not use_dora else "dora"
     for i, image in enumerate(images):
         image.save(os.path.join(repo_folder, f"image_{i}.png"))
         img_str += f"""
@@ -139,9 +141,10 @@ def save_model_card(
 tags:
 - stable-diffusion
 - stable-diffusion-diffusers
+- diffusers-training
 - text-to-image
 - diffusers
-- lora
+- {lora}
 - template:sd-lora
 {img_str}
 base_model: {base_model}
@@ -1967,6 +1970,7 @@ def compute_text_embeddings(prompt, text_encoders, tokenizers):
 
         save_model_card(
             model_id if not args.push_to_hub else repo_id,
+            use_dora=args.use_dora,
             images=images,
             base_model=args.pretrained_model_name_or_path,
             train_text_encoder=args.train_text_encoder,
diff --git a/examples/advanced_diffusion_training/train_dreambooth_lora_sdxl_advanced.py b/examples/advanced_diffusion_training/train_dreambooth_lora_sdxl_advanced.py
@@ -81,6 +81,7 @@
 
 def save_model_card(
     repo_id: str,
+    use_dora: bool,
     images=None,
     base_model=str,
     train_text_encoder=False,
@@ -92,6 +93,7 @@ def save_model_card(
     vae_path=None,
 ):
     img_str = "widget:\n"
+    lora = "lora" if not use_dora else "dora"
     for i, image in enumerate(images):
         image.save(os.path.join(repo_folder, f"image_{i}.png"))
         img_str += f"""
@@ -144,9 +146,10 @@ def save_model_card(
 tags:
 - stable-diffusion-xl
 - stable-diffusion-xl-diffusers
+- diffusers-training
 - text-to-image
 - diffusers
-- lora
+- {lora}
 - template:sd-lora
 {img_str}
 base_model: {base_model}
@@ -2204,6 +2207,7 @@ def compute_text_embeddings(prompt, text_encoders, tokenizers):
 
         save_model_card(
             model_id if not args.push_to_hub else repo_id,
+            use_dora=args.use_dora,
             images=images,
             base_model=args.pretrained_model_name_or_path,
             train_text_encoder=args.train_text_encoder,
diff --git a/examples/controlnet/train_controlnet.py b/examples/controlnet/train_controlnet.py
@@ -242,6 +242,7 @@ def save_model_card(repo_id: str, image_logs=None, base_model=str, repo_folder=N
         "text-to-image",
         "diffusers",
         "controlnet",
+        "diffusers-training",
     ]
     model_card = populate_model_card(model_card, tags=tags)
 
diff --git a/examples/controlnet/train_controlnet_flax.py b/examples/controlnet/train_controlnet_flax.py
@@ -169,6 +169,7 @@ def save_model_card(repo_id: str, image_logs=None, base_model=str, repo_folder=N
         "diffusers",
         "controlnet",
         "jax-diffusers-event",
+        "diffusers-training",
     ]
     model_card = populate_model_card(model_card, tags=tags)
 
diff --git a/examples/controlnet/train_controlnet_sdxl.py b/examples/controlnet/train_controlnet_sdxl.py
@@ -243,6 +243,7 @@ def save_model_card(repo_id: str, image_logs=None, base_model=str, repo_folder=N
         "text-to-image",
         "diffusers",
         "controlnet",
+        "diffusers-training",
     ]
     model_card = populate_model_card(model_card, tags=tags)
 
diff --git a/examples/custom_diffusion/train_custom_diffusion.py b/examples/custom_diffusion/train_custom_diffusion.py
@@ -97,7 +97,14 @@ def save_model_card(repo_id: str, images=None, base_model=str, prompt=str, repo_
         inference=True,
     )
 
-    tags = ["text-to-image", "diffusers", "stable-diffusion", "stable-diffusion-diffusers", "custom-diffusion"]
+    tags = [
+        "text-to-image",
+        "diffusers",
+        "stable-diffusion",
+        "stable-diffusion-diffusers",
+        "custom-diffusion",
+        "diffusers-training",
+    ]
     model_card = populate_model_card(model_card, tags=tags)
 
     model_card.save(os.path.join(repo_folder, "README.md"))
diff --git a/examples/dreambooth/train_dreambooth.py b/examples/dreambooth/train_dreambooth.py
@@ -102,7 +102,7 @@ def save_model_card(
         inference=True,
     )
 
-    tags = ["text-to-image", "dreambooth"]
+    tags = ["text-to-image", "dreambooth", "diffusers-training"]
     if isinstance(pipeline, StableDiffusionPipeline):
         tags.extend(["stable-diffusion", "stable-diffusion-diffusers"])
     else:
diff --git a/examples/dreambooth/train_dreambooth_lora.py b/examples/dreambooth/train_dreambooth_lora.py
@@ -106,7 +106,7 @@ def save_model_card(
         model_description=model_description,
         inference=True,
     )
-    tags = ["text-to-image", "diffusers", "lora"]
+    tags = ["text-to-image", "diffusers", "lora", "diffusers-training"]
     if isinstance(pipeline, StableDiffusionPipeline):
         tags.extend(["stable-diffusion", "stable-diffusion-diffusers"])
     else:
diff --git a/examples/dreambooth/train_dreambooth_lora_sdxl.py b/examples/dreambooth/train_dreambooth_lora_sdxl.py
@@ -96,6 +96,7 @@ def determine_scheduler_type(pretrained_model_name_or_path, revision):
 
 def save_model_card(
     repo_id: str,
+    use_dora: bool,
     images=None,
     base_model: str = None,
     train_text_encoder=False,
@@ -156,8 +157,9 @@ def save_model_card(
     tags = [
         "text-to-image",
         "text-to-image",
+        "diffusers-training",
         "diffusers",
-        "lora",
+        "lora" if not use_dora else "dora",
         "template:sd-lora",
     ]
     if "playgroundai" in base_model:
@@ -1912,6 +1914,7 @@ def get_sigmas(timesteps, n_dim=4, dtype=torch.float32):
         if args.push_to_hub:
             save_model_card(
                 repo_id,
+                use_dora=args.use_dora,
                 images=images,
                 base_model=args.pretrained_model_name_or_path,
                 train_text_encoder=args.train_text_encoder,
diff --git a/examples/kandinsky2_2/text_to_image/train_text_to_image_decoder.py b/examples/kandinsky2_2/text_to_image/train_text_to_image_decoder.py
@@ -81,6 +81,7 @@ def save_model_card(
 - kandinsky
 - text-to-image
 - diffusers
+- diffusers-training
 inference: true
 ---
     """
diff --git a/examples/kandinsky2_2/text_to_image/train_text_to_image_lora_decoder.py b/examples/kandinsky2_2/text_to_image/train_text_to_image_lora_decoder.py
@@ -65,6 +65,7 @@ def save_model_card(repo_id: str, images=None, base_model=str, dataset_name=str,
 - kandinsky
 - text-to-image
 - diffusers
+- diffusers-training
 - lora
 inference: true
 ---
diff --git a/examples/kandinsky2_2/text_to_image/train_text_to_image_lora_prior.py b/examples/kandinsky2_2/text_to_image/train_text_to_image_lora_prior.py
@@ -65,6 +65,7 @@ def save_model_card(repo_id: str, images=None, base_model=str, dataset_name=str,
 - kandinsky
 - text-to-image
 - diffusers
+- diffusers-training
 - lora
 inference: true
 ---
diff --git a/examples/kandinsky2_2/text_to_image/train_text_to_image_prior.py b/examples/kandinsky2_2/text_to_image/train_text_to_image_prior.py
@@ -82,6 +82,7 @@ def save_model_card(
 - kandinsky
 - text-to-image
 - diffusers
+- diffusers-training
 inference: true
 ---
     """
diff --git a/examples/research_projects/controlnet/train_controlnet_webdataset.py b/examples/research_projects/controlnet/train_controlnet_webdataset.py
@@ -460,6 +460,8 @@ def save_model_card(repo_id: str, image_logs=None, base_model=str, repo_folder=N
 - text-to-image
 - diffusers
 - controlnet
+- diffusers-training
+- webdataset
 inference: true
 ---
     """
diff --git a/examples/research_projects/lora/train_text_to_image_lora.py b/examples/research_projects/lora/train_text_to_image_lora.py
@@ -69,6 +69,7 @@ def save_model_card(repo_id: str, images=None, base_model=str, dataset_name=str,
 - stable-diffusion-diffusers
 - text-to-image
 - diffusers
+- diffusers-training
 - lora
 inference: true
 ---
diff --git a/examples/research_projects/onnxruntime/textual_inversion/textual_inversion.py b/examples/research_projects/onnxruntime/textual_inversion/textual_inversion.py
@@ -100,6 +100,8 @@ def save_model_card(repo_id: str, images=None, base_model=str, repo_folder=None)
 - text-to-image
 - diffusers
 - textual_inversion
+- diffusers-training
+- onxruntime
 inference: true
 ---
     """
diff --git a/examples/research_projects/realfill/train_realfill.py b/examples/research_projects/realfill/train_realfill.py
@@ -87,6 +87,7 @@ def save_model_card(
 - text-to-image
 - diffusers
 - realfill
+- diffusers-training
 inference: true
 ---
     """
diff --git a/examples/t2i_adapter/train_t2i_adapter_sdxl.py b/examples/t2i_adapter/train_t2i_adapter_sdxl.py
@@ -225,7 +225,14 @@ def save_model_card(repo_id: str, image_logs: dict = None, base_model: str = Non
         inference=True,
     )
 
-    tags = ["stable-diffusion-xl", "stable-diffusion-xl-diffusers", "text-to-image", "diffusers", "t2iadapter"]
+    tags = [
+        "stable-diffusion-xl",
+        "stable-diffusion-xl-diffusers",
+        "text-to-image",
+        "diffusers",
+        "t2iadapter",
+        "diffusers-training",
+    ]
     model_card = populate_model_card(model_card, tags=tags)
 
     model_card.save(os.path.join(repo_folder, "README.md"))
diff --git a/examples/text_to_image/train_text_to_image.py b/examples/text_to_image/train_text_to_image.py
@@ -131,7 +131,7 @@ def save_model_card(
         inference=True,
     )
 
-    tags = ["stable-diffusion", "stable-diffusion-diffusers", "text-to-image", "diffusers"]
+    tags = ["stable-diffusion", "stable-diffusion-diffusers", "text-to-image", "diffusers", "diffusers-training"]
     model_card = populate_model_card(model_card, tags=tags)
 
     model_card.save(os.path.join(repo_folder, "README.md"))
diff --git a/examples/text_to_image/train_text_to_image_lora.py b/examples/text_to_image/train_text_to_image_lora.py
@@ -90,6 +90,7 @@ def save_model_card(
         "stable-diffusion-diffusers",
         "text-to-image",
         "diffusers",
+        "diffusers-training",
         "lora",
     ]
     model_card = populate_model_card(model_card, tags=tags)
diff --git a/examples/text_to_image/train_text_to_image_lora_sdxl.py b/examples/text_to_image/train_text_to_image_lora_sdxl.py
@@ -103,7 +103,14 @@ def save_model_card(
         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.save(os.path.join(repo_folder, "README.md"))
diff --git a/examples/text_to_image/train_text_to_image_sdxl.py b/examples/text_to_image/train_text_to_image_sdxl.py
@@ -101,6 +101,7 @@ def save_model_card(
         "stable-diffusion-xl",
         "stable-diffusion-xl-diffusers",
         "text-to-image",
+        "diffusers-training",
         "diffusers",
     ]
     model_card = populate_model_card(model_card, tags=tags)
diff --git a/examples/textual_inversion/textual_inversion.py b/examples/textual_inversion/textual_inversion.py
@@ -105,7 +105,14 @@ def save_model_card(repo_id: str, images: list = None, base_model: str = None, r
         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.save(os.path.join(repo_folder, "README.md"))
diff --git a/examples/textual_inversion/textual_inversion_sdxl.py b/examples/textual_inversion/textual_inversion_sdxl.py
@@ -106,6 +106,7 @@ def save_model_card(repo_id: str, images=None, base_model=str, repo_folder=None)
         "stable-diffusion-xl-diffusers",
         "text-to-image",
         "diffusers",
+        "diffusers-training",
         "textual_inversion",
     ]
 
diff --git a/examples/wuerstchen/text_to_image/train_text_to_image_lora_prior.py b/examples/wuerstchen/text_to_image/train_text_to_image_lora_prior.py
@@ -81,6 +81,7 @@ def save_model_card(
 - wuerstchen
 - text-to-image
 - diffusers
+- diffusers-training
 - lora
 inference: true
 ---
diff --git a/examples/wuerstchen/text_to_image/train_text_to_image_prior.py b/examples/wuerstchen/text_to_image/train_text_to_image_prior.py
@@ -82,6 +82,7 @@ def save_model_card(
 - wuerstchen
 - text-to-image
 - diffusers
+- diffusers-training
 inference: true
 ---
     """

Original file line number	Diff line number	Diff line change
`@@ -242,6 +242,7 @@ def save_model_card(repo_id: str, image_logs=None, base_model=str, repo_folder=N`
`242`	`242`	`"text-to-image",`
`243`	`243`	`"diffusers",`
`244`	`244`	`"controlnet",`
	`245`	`+ "diffusers-training",`
`245`	`246`	`]`
`246`	`247`	`model_card = populate_model_card(model_card, tags=tags)`
`247`	`248`
Original file line number	Diff line number	Diff line change
`@@ -169,6 +169,7 @@ def save_model_card(repo_id: str, image_logs=None, base_model=str, repo_folder=N`
`169`	`169`	`"diffusers",`
`170`	`170`	`"controlnet",`
`171`	`171`	`"jax-diffusers-event",`
	`172`	`+ "diffusers-training",`
`172`	`173`	`]`
`173`	`174`	`model_card = populate_model_card(model_card, tags=tags)`
`174`	`175`
Original file line number	Diff line number	Diff line change
`@@ -243,6 +243,7 @@ def save_model_card(repo_id: str, image_logs=None, base_model=str, repo_folder=N`
`243`	`243`	`"text-to-image",`
`244`	`244`	`"diffusers",`
`245`	`245`	`"controlnet",`
	`246`	`+ "diffusers-training",`
`246`	`247`	`]`
`247`	`248`	`model_card = populate_model_card(model_card, tags=tags)`
`248`	`249`
Original file line number	Diff line number	Diff line change
`@@ -102,7 +102,7 @@ def save_model_card(`
`102`	`102`	`inference=True,`
`103`	`103`	`)`
`104`	`104`
`105`		`- tags = ["text-to-image", "dreambooth"]`
	`105`	`+ tags = ["text-to-image", "dreambooth", "diffusers-training"]`
`106`	`106`	`if isinstance(pipeline, StableDiffusionPipeline):`
`107`	`107`	`tags.extend(["stable-diffusion", "stable-diffusion-diffusers"])`
`108`	`108`	`else:`
Original file line number	Diff line number	Diff line change
`@@ -106,7 +106,7 @@ def save_model_card(`
`106`	`106`	`model_description=model_description,`
`107`	`107`	`inference=True,`
`108`	`108`	`)`
`109`		`- tags = ["text-to-image", "diffusers", "lora"]`
	`109`	`+ tags = ["text-to-image", "diffusers", "lora", "diffusers-training"]`
`110`	`110`	`if isinstance(pipeline, StableDiffusionPipeline):`
`111`	`111`	`tags.extend(["stable-diffusion", "stable-diffusion-diffusers"])`
`112`	`112`	`else:`