Skip to content

Commit e53ec15

Browse files
authored
Merge branch 'main' into fix-leaf-offload-bug-12396
2 parents 09456cb + eeae033 commit e53ec15

File tree

85 files changed

+5635
-656
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

85 files changed

+5635
-656
lines changed

docs/source/en/_toctree.yml

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,8 @@
2222
title: Reproducibility
2323
- local: using-diffusers/schedulers
2424
title: Schedulers
25+
- local: using-diffusers/automodel
26+
title: AutoModel
2527
- local: using-diffusers/other-formats
2628
title: Model formats
2729
- local: using-diffusers/push_to_hub
@@ -119,6 +121,8 @@
119121
title: ComponentsManager
120122
- local: modular_diffusers/guiders
121123
title: Guiders
124+
- local: modular_diffusers/custom_blocks
125+
title: Building Custom Blocks
122126
title: Modular Diffusers
123127
- isExpanded: false
124128
sections:
@@ -387,6 +391,8 @@
387391
title: Transformer2DModel
388392
- local: api/models/transformer_temporal
389393
title: TransformerTemporalModel
394+
- local: api/models/wan_animate_transformer_3d
395+
title: WanAnimateTransformer3DModel
390396
- local: api/models/wan_transformer_3d
391397
title: WanTransformer3DModel
392398
title: Transformers
@@ -448,6 +454,8 @@
448454
- sections:
449455
- local: api/pipelines/overview
450456
title: Overview
457+
- local: api/pipelines/auto_pipeline
458+
title: AutoPipeline
451459
- sections:
452460
- local: api/pipelines/audioldm
453461
title: AudioLDM
@@ -460,8 +468,6 @@
460468
- local: api/pipelines/stable_audio
461469
title: Stable Audio
462470
title: Audio
463-
- local: api/pipelines/auto_pipeline
464-
title: AutoPipeline
465471
- sections:
466472
- local: api/pipelines/amused
467473
title: aMUSEd
@@ -525,6 +531,8 @@
525531
title: HiDream-I1
526532
- local: api/pipelines/hunyuandit
527533
title: Hunyuan-DiT
534+
- local: api/pipelines/hunyuanimage21
535+
title: HunyuanImage2.1
528536
- local: api/pipelines/pix2pix
529537
title: InstructPix2Pix
530538
- local: api/pipelines/kandinsky
@@ -638,8 +646,6 @@
638646
title: ConsisID
639647
- local: api/pipelines/framepack
640648
title: Framepack
641-
- local: api/pipelines/hunyuanimage21
642-
title: HunyuanImage2.1
643649
- local: api/pipelines/hunyuan_video
644650
title: HunyuanVideo
645651
- local: api/pipelines/i2vgenxl

docs/source/en/api/models/auto_model.md

Lines changed: 1 addition & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -12,15 +12,7 @@ specific language governing permissions and limitations under the License.
1212

1313
# AutoModel
1414

15-
The `AutoModel` is designed to make it easy to load a checkpoint without needing to know the specific model class. `AutoModel` automatically retrieves the correct model class from the checkpoint `config.json` file.
16-
17-
```python
18-
from diffusers import AutoModel, AutoPipelineForText2Image
19-
20-
unet = AutoModel.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5", subfolder="unet")
21-
pipe = AutoPipelineForText2Image.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5", unet=unet)
22-
```
23-
15+
[`AutoModel`] automatically retrieves the correct model class from the checkpoint `config.json` file.
2416

2517
## AutoModel
2618

Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
1+
<!-- Copyright 2025 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+
# WanAnimateTransformer3DModel
13+
14+
A Diffusion Transformer model for 3D video-like data was introduced in [Wan Animate](https://github.com/Wan-Video/Wan2.2) by the Alibaba Wan Team.
15+
16+
The model can be loaded with the following code snippet.
17+
18+
```python
19+
from diffusers import WanAnimateTransformer3DModel
20+
21+
transformer = WanAnimateTransformer3DModel.from_pretrained("Wan-AI/Wan2.2-Animate-14B-720P-Diffusers", subfolder="transformer", torch_dtype=torch.bfloat16)
22+
```
23+
24+
## WanAnimateTransformer3DModel
25+
26+
[[autodoc]] WanAnimateTransformer3DModel
27+
28+
## Transformer2DModelOutput
29+
30+
[[autodoc]] models.modeling_outputs.Transformer2DModelOutput

0 commit comments

Comments
 (0)