Skip to content

Commit 4704586

Browse files
authored
[docs] fix single_file example. (#11847)
fix single_file example.
1 parent a79c3af commit 4704586

File tree

1 file changed

+3
-3
lines changed
  • docs/source/en/api/pipelines

1 file changed

+3
-3
lines changed

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

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -302,12 +302,12 @@ The general rule of thumb to keep in mind when preparing inputs for the VACE pip
302302
```py
303303
# pip install ftfy
304304
import torch
305-
from diffusers import WanPipeline, AutoModel
305+
from diffusers import WanPipeline, WanTransformer3DModel, AutoencoderKLWan
306306

307-
vae = AutoModel.from_single_file(
307+
vae = AutoencoderKLWan.from_single_file(
308308
"https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/blob/main/split_files/vae/wan_2.1_vae.safetensors"
309309
)
310-
transformer = AutoModel.from_single_file(
310+
transformer = WanTransformer3DModel.from_single_file(
311311
"https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/blob/main/split_files/diffusion_models/wan2.1_t2v_1.3B_bf16.safetensors",
312312
torch_dtype=torch.bfloat16
313313
)

0 commit comments

Comments
 (0)