Skip to content

Commit 731b3bb

Browse files
committed
Merge branch 'to-single-file/flux' into attn-dispatcher-cp-and-training
2 parents 62f164d + 9f37b87 commit 731b3bb

31 files changed

+7716
-4
lines changed

docs/source/en/_toctree.yml

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -353,6 +353,8 @@
353353
title: SanaTransformer2DModel
354354
- local: api/models/sd3_transformer2d
355355
title: SD3Transformer2DModel
356+
- local: api/models/skyreels_v2_transformer_3d
357+
title: SkyReelsV2Transformer3DModel
356358
- local: api/models/stable_audio_transformer
357359
title: StableAudioDiTModel
358360
- local: api/models/transformer2d
@@ -547,6 +549,8 @@
547549
title: Semantic Guidance
548550
- local: api/pipelines/shap_e
549551
title: Shap-E
552+
- local: api/pipelines/skyreels_v2
553+
title: SkyReels-V2
550554
- local: api/pipelines/stable_audio
551555
title: Stable Audio
552556
- local: api/pipelines/stable_cascade

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

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@ LoRA is a fast and lightweight training method that inserts and trains a signifi
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).
2828
- [`WanLoraLoaderMixin`] provides similar functions for [Wan](https://huggingface.co/docs/diffusers/main/en/api/pipelines/wan).
29+
- [`SkyReelsV2LoraLoaderMixin`] provides similar functions for [SkyReels-V2](https://huggingface.co/docs/diffusers/main/en/api/pipelines/skyreels_v2).
2930
- [`CogView4LoraLoaderMixin`] provides similar functions for [CogView4](https://huggingface.co/docs/diffusers/main/en/api/pipelines/cogview4).
3031
- [`AmusedLoraLoaderMixin`] is for the [`AmusedPipeline`].
3132
- [`HiDreamImageLoraLoaderMixin`] provides similar functions for [HiDream Image](https://huggingface.co/docs/diffusers/main/en/api/pipelines/hidream)
@@ -92,6 +93,10 @@ To learn more about how to load LoRA weights, see the [LoRA](../../using-diffuse
9293

9394
[[autodoc]] loaders.lora_pipeline.WanLoraLoaderMixin
9495

96+
## SkyReelsV2LoraLoaderMixin
97+
98+
[[autodoc]] loaders.lora_pipeline.SkyReelsV2LoraLoaderMixin
99+
95100
## AmusedLoraLoaderMixin
96101

97102
[[autodoc]] loaders.lora_pipeline.AmusedLoraLoaderMixin
@@ -100,6 +105,6 @@ To learn more about how to load LoRA weights, see the [LoRA](../../using-diffuse
100105

101106
[[autodoc]] loaders.lora_pipeline.HiDreamImageLoraLoaderMixin
102107

103-
## WanLoraLoaderMixin
108+
## LoraBaseMixin
104109

105-
[[autodoc]] loaders.lora_pipeline.WanLoraLoaderMixin
110+
[[autodoc]] loaders.lora_base.LoraBaseMixin
Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
1+
<!-- Copyright 2024 The HuggingFace Team. All rights reserved.
2+
3+
Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
4+
the License. You may obtain a copy of the License at
5+
6+
http://www.apache.org/licenses/LICENSE-2.0
7+
8+
Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
9+
an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
10+
specific language governing permissions and limitations under the License. -->
11+
12+
# SkyReelsV2Transformer3DModel
13+
14+
A Diffusion Transformer model for 3D video-like data was introduced in [SkyReels-V2](https://github.com/SkyworkAI/SkyReels-V2) by the Skywork AI.
15+
16+
The model can be loaded with the following code snippet.
17+
18+
```python
19+
from diffusers import SkyReelsV2Transformer3DModel
20+
21+
transformer = SkyReelsV2Transformer3DModel.from_pretrained("Skywork/SkyReels-V2-DF-1.3B-540P-Diffusers", subfolder="transformer", torch_dtype=torch.bfloat16)
22+
```
23+
24+
## SkyReelsV2Transformer3DModel
25+
26+
[[autodoc]] SkyReelsV2Transformer3DModel
27+
28+
## Transformer2DModelOutput
29+
30+
[[autodoc]] models.modeling_outputs.Transformer2DModelOutput

docs/source/en/api/pipelines/skyreels_v2.md

Lines changed: 367 additions & 0 deletions
Large diffs are not rendered by default.

0 commit comments

Comments
 (0)