Skip to content

Commit efc9d68

Browse files
authored
[chore] fix lora docs utils (#11338)
fix lora docs utils
1 parent 3e59d53 commit efc9d68

File tree

2 files changed

+11
-1
lines changed

2 files changed

+11
-1
lines changed

docs/source/en/api/loaders/lora.md

+10
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,8 @@ LoRA is a fast and lightweight training method that inserts and trains a signifi
2525
- [`SanaLoraLoaderMixin`] provides similar functions for [Sana](https://huggingface.co/docs/diffusers/main/en/api/pipelines/sana).
2626
- [`HunyuanVideoLoraLoaderMixin`] provides similar functions for [HunyuanVideo](https://huggingface.co/docs/diffusers/main/en/api/pipelines/hunyuan_video).
2727
- [`Lumina2LoraLoaderMixin`] provides similar functions for [Lumina2](https://huggingface.co/docs/diffusers/main/en/api/pipelines/lumina2).
28+
- [`WanLoraLoaderMixin`] provides similar functions for [Wan](https://huggingface.co/docs/diffusers/main/en/api/pipelines/wan).
29+
- [`CogView4LoraLoaderMixin`] provides similar functions for [CogView4](https://huggingface.co/docs/diffusers/main/en/api/pipelines/cogview4).
2830
- [`AmusedLoraLoaderMixin`] is for the [`AmusedPipeline`].
2931
- [`LoraBaseMixin`] provides a base class with several utility methods to fuse, unfuse, unload, LoRAs and more.
3032

@@ -77,6 +79,14 @@ To learn more about how to load LoRA weights, see the [LoRA](../../using-diffuse
7779

7880
[[autodoc]] loaders.lora_pipeline.Lumina2LoraLoaderMixin
7981

82+
## CogView4LoraLoaderMixin
83+
84+
[[autodoc]] loaders.lora_pipeline.CogView4LoraLoaderMixin
85+
86+
## WanLoraLoaderMixin
87+
88+
[[autodoc]] loaders.lora_pipeline.WanLoraLoaderMixin
89+
8090
## AmusedLoraLoaderMixin
8191

8292
[[autodoc]] loaders.lora_pipeline.AmusedLoraLoaderMixin

utils/check_support_list.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -100,7 +100,7 @@ def check_documentation(doc_path, src_path, doc_regex, src_regex, exclude_condit
100100
"doc_path": "docs/source/en/api/loaders/lora.md",
101101
"src_path": "src/diffusers/loaders/lora_pipeline.py",
102102
"doc_regex": r"\[\[autodoc\]\]\s([^\n]+)",
103-
"src_regex": r"class\s+(\w+)\s*\(.*?nn\.Module.*?\):",
103+
"src_regex": r"class\s+(\w+LoraLoaderMixin(?:\d*_?\d*))[:(]",
104104
},
105105
}
106106

0 commit comments

Comments
 (0)