-
Notifications
You must be signed in to change notification settings - Fork 2
Description
python generate.py
--task s2v-5B
--ckpt_dir ./ckpts/Wan2.2-TI2V-5B
--lora_ckpt ./ckpts/talkverse_5b_lora.safetensors
--image examples/input_face.jpg
--audio examples/input_audio.wav
--prompt "A person talking naturally."
--offload_model True
--t5_cpu
i change the image and audio to my local path, but something error occurs
Traceback (most recent call last):
File "/dh-nas-dev/tujiexu/workplace/code/cv/TalkVerse/generate.py", line 779, in
generate(args)
~~~~~~~~^^^^^^
File "/dh-nas-dev/tujiexu/workplace/code/cv/TalkVerse/generate.py", line 649, in generate
wan_s2v = wan.WanS2V_5B(
config=cfg,
...<7 lines>...
convert_model_dtype=args.convert_model_dtype,
)
File "/dh-nas-dev/tujiexu/workplace/code/cv/TalkVerse/wan/s2v_5b.py", line 353, in init
self._load_pretrained_weights(checkpoint_dir)
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^
File "/dh-nas-dev/tujiexu/workplace/code/cv/TalkVerse/wan/s2v_5b.py", line 436, in _load_pretrained_weights
raise NotImplementedError
NotImplementedError