mirror of
https://github.com/huggingface/diffusers.git
synced 2025-12-15 08:54:20 +08:00
Compare commits
1 Commits
push-test-
...
misc-docs-
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
b5ba24b1dc |
3
.github/workflows/push_tests.yml
vendored
3
.github/workflows/push_tests.yml
vendored
@@ -165,8 +165,7 @@ jobs:
|
|||||||
group: gcp-ct5lp-hightpu-8t
|
group: gcp-ct5lp-hightpu-8t
|
||||||
container:
|
container:
|
||||||
image: diffusers/diffusers-flax-tpu
|
image: diffusers/diffusers-flax-tpu
|
||||||
options: --shm-size "16gb" --ipc host --privileged ${{ vars.V5_LITEPOD_8_ENV}} -v /mnt/hf_cache:/mnt/hf_cache
|
options: --shm-size "16gb" --ipc host --privileged ${{ vars.V5_LITEPOD_8_ENV}} -v /mnt/hf_cache:/mnt/hf_cache defaults:
|
||||||
defaults:
|
|
||||||
run:
|
run:
|
||||||
shell: bash
|
shell: bash
|
||||||
steps:
|
steps:
|
||||||
|
|||||||
@@ -17,6 +17,9 @@ LoRA is a fast and lightweight training method that inserts and trains a signifi
|
|||||||
- [`StableDiffusionLoraLoaderMixin`] provides functions for loading and unloading, fusing and unfusing, enabling and disabling, and more functions for managing LoRA weights. This class can be used with any model.
|
- [`StableDiffusionLoraLoaderMixin`] provides functions for loading and unloading, fusing and unfusing, enabling and disabling, and more functions for managing LoRA weights. This class can be used with any model.
|
||||||
- [`StableDiffusionXLLoraLoaderMixin`] is a [Stable Diffusion (SDXL)](../../api/pipelines/stable_diffusion/stable_diffusion_xl) version of the [`StableDiffusionLoraLoaderMixin`] class for loading and saving LoRA weights. It can only be used with the SDXL model.
|
- [`StableDiffusionXLLoraLoaderMixin`] is a [Stable Diffusion (SDXL)](../../api/pipelines/stable_diffusion/stable_diffusion_xl) version of the [`StableDiffusionLoraLoaderMixin`] class for loading and saving LoRA weights. It can only be used with the SDXL model.
|
||||||
- [`SD3LoraLoaderMixin`] provides similar functions for [Stable Diffusion 3](https://huggingface.co/blog/sd3).
|
- [`SD3LoraLoaderMixin`] provides similar functions for [Stable Diffusion 3](https://huggingface.co/blog/sd3).
|
||||||
|
- [`FluxLoraLoaderMixin`] provides similar functions for [Flux](https://huggingface.co/docs/diffusers/main/en/api/pipelines/flux).
|
||||||
|
- [`CogVideoXLoraLoaderMixin`] provides similar functions for [CogVideoX](https://huggingface.co/docs/diffusers/main/en/api/pipelines/cogvideox).
|
||||||
|
- [`Mochi1LoraLoaderMixin`] provides similar functions for [Mochi](https://huggingface.co/docs/diffusers/main/en/api/pipelines/mochi).
|
||||||
- [`AmusedLoraLoaderMixin`] is for the [`AmusedPipeline`].
|
- [`AmusedLoraLoaderMixin`] is for the [`AmusedPipeline`].
|
||||||
- [`LoraBaseMixin`] provides a base class with several utility methods to fuse, unfuse, unload, LoRAs and more.
|
- [`LoraBaseMixin`] provides a base class with several utility methods to fuse, unfuse, unload, LoRAs and more.
|
||||||
|
|
||||||
@@ -38,6 +41,18 @@ To learn more about how to load LoRA weights, see the [LoRA](../../using-diffuse
|
|||||||
|
|
||||||
[[autodoc]] loaders.lora_pipeline.SD3LoraLoaderMixin
|
[[autodoc]] loaders.lora_pipeline.SD3LoraLoaderMixin
|
||||||
|
|
||||||
|
## FluxLoraLoaderMixin
|
||||||
|
|
||||||
|
[[autodoc]] loaders.lora_pipeline.FluxLoraLoaderMixin
|
||||||
|
|
||||||
|
## CogVideoXLoraLoaderMixin
|
||||||
|
|
||||||
|
[[autodoc]] loaders.lora_pipeline.CogVideoXLoraLoaderMixin
|
||||||
|
|
||||||
|
## Mochi1LoraLoaderMixin
|
||||||
|
|
||||||
|
[[autodoc]] loaders.lora_pipeline.Mochi1LoraLoaderMixin
|
||||||
|
|
||||||
## AmusedLoraLoaderMixin
|
## AmusedLoraLoaderMixin
|
||||||
|
|
||||||
[[autodoc]] loaders.lora_pipeline.AmusedLoraLoaderMixin
|
[[autodoc]] loaders.lora_pipeline.AmusedLoraLoaderMixin
|
||||||
|
|||||||
@@ -26,7 +26,7 @@ Make sure to check out the Schedulers [guide](../../using-diffusers/schedulers.m
|
|||||||
|
|
||||||
</Tip>
|
</Tip>
|
||||||
|
|
||||||
This pipeline was contributed by [lawrence-cj](https://github.com/lawrence-cj) and [chenjy2003](https://github.com/chenjy2003). The original codebase can be found [here](https://github.com/NVlabs/Sana). The original weights can be found under [hf.co/Efficient-Large-Model](https://huggingface.co/Efficient-Large-Model).
|
This pipeline was contributed by [lawrence-cj](https://github.com/lawrence-cj). The original codebase can be found [here](https://github.com/NVlabs/Sana). The original weights can be found under [hf.co/Efficient-Large-Model]https://huggingface.co/Efficient-Large-Model).
|
||||||
|
|
||||||
Available models:
|
Available models:
|
||||||
|
|
||||||
|
|||||||
@@ -16,8 +16,12 @@ ALWAYS_TEST_PIPELINE_MODULES = [
|
|||||||
"stable_diffusion_2",
|
"stable_diffusion_2",
|
||||||
"stable_diffusion_xl",
|
"stable_diffusion_xl",
|
||||||
"stable_diffusion_adapter",
|
"stable_diffusion_adapter",
|
||||||
|
"deepfloyd_if",
|
||||||
"ip_adapters",
|
"ip_adapters",
|
||||||
|
"kandinsky",
|
||||||
"kandinsky2_2",
|
"kandinsky2_2",
|
||||||
|
"text_to_video_synthesis",
|
||||||
|
"wuerstchen",
|
||||||
]
|
]
|
||||||
PIPELINE_USAGE_CUTOFF = int(os.getenv("PIPELINE_USAGE_CUTOFF", 50000))
|
PIPELINE_USAGE_CUTOFF = int(os.getenv("PIPELINE_USAGE_CUTOFF", 50000))
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user