File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
docs/source/en/api/pipelines Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff 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 )
You can’t perform that action at this time.
0 commit comments