Skip to content

GitLab

Unverified Commit 47045862 authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

[docs] fix single_file example. (#11847) 

fix single_file example.
parent a79c3af6
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
docs/source/en/api/pipelines/wan.md
View file @ 47045862
...@@ -302,12 +302,12 @@ The general rule of thumb to keep in mind when preparing inputs for the VACE pip ...@@ -302,12 +302,12 @@ The general rule of thumb to keep in mind when preparing inputs for the VACE pip
```py ```py
# pip install ftfy # pip install ftfy
import torch import torch
from diffusers import WanPipeline, AutoModel from diffusers import WanPipeline, WanTransformer3DModel, AutoencoderKLWan
vae = AutoModel.from_single_file( vae = AutoencoderKLWan.from_single_file(
"https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/blob/main/split_files/vae/wan_2.1_vae.safetensors" "https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/blob/main/split_files/vae/wan_2.1_vae.safetensors"
) )
transformer = AutoModel.from_single_file( transformer = WanTransformer3DModel.from_single_file(
"https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/blob/main/split_files/diffusion_models/wan2.1_t2v_1.3B_bf16.safetensors", "https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/blob/main/split_files/diffusion_models/wan2.1_t2v_1.3B_bf16.safetensors",
torch_dtype=torch.bfloat16 torch_dtype=torch.bfloat16
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment