load wan2.2 animate from local path error #1028
Description
Hi, I just trying to load animate model from my local path, then got errors as following:
import torch, os, json
from diffsynth import load_state_dict
from diffsynth.pipelines.wan_video_new import WanVideoPipeline, ModelConfig
from diffsynth.trainers.utils import DiffusionTrainingModule, ModelLogger, launch_training_task, wan_parser
from diffsynth.trainers.unified_dataset import UnifiedDataset, LoadVideo, LoadAudio, ImageCropAndResize, ToAbsolutePath
os.environ["TOKENIZERS_PARALLELISM"] = "false"
parent_path = './models'
pipe = WanVideoPipeline.from_pretrained(
torch_dtype=torch.bfloat16,
device="cuda",
model_configs=[
ModelConfig(path=[
f"{parent_path}/Wan2.2-Animate-14B/models_t5_umt5-xxl-enc-bf16.pth",
], offload_device="cpu"),
ModelConfig(path=[
f"{parent_path}/Wan2.2-Animate-14B/Wan2.2_VAE.pth"
], offload_device="cpu"),
ModelConfig(path=[
f"{parent_path}/Wan2.2-Animate-14B/diffusion_pytorch_model-00001-of-00004.safetensors",
f"{parent_path}/Wan2.2-Animate-14B/diffusion_pytorch_model-00002-of-00004.safetensors",
f"{parent_path}/Wan2.2-Animate-14B/diffusion_pytorch_model-00003-of-00004.safetensors",
f"{parent_path}/Wan2.2-Animate-14B/diffusion_pytorch_model-00004-of-00004.safetensors",
], offload_device="cpu"),
ModelConfig(path=[
f"{parent_path}/Wan2.2-Animate-14B/models_clip_open-clip-xlm-roberta-large-vit-huge-14.pth"
], offload_device="cpu"),
],
)
pipe.enable_vram_management()
error is
No wan_video_motion_controller models available.
No wan_video_vace models available.
No wans2v_audio_encoder models available.