Skip to content

按官方的配置部署完成后,运行不起来呀?是不是配置不对呢? #207

@sandheart

Description

@sandheart

根据官方配置在windows11上总卡在pip install flash_attn==2.7.4.post1“”,用wsl部署能成功完成,但运行时失,提示如下:
python generate_infinitetalk.py
--ckpt_dir weights/Wan2.1-I2V-14B-480P
--wav2vec_dir 'weights/chinese-wav2vec2-base'
--infinitetalk_dir weights/InfiniteTalk/single/infinitetalk.safetensors
--input_json examples/single_example_image.json
--size infinitetalk-480
--sample_steps 40
--mode streaming
--motion_frame 9
--save_file infinitetalk_res
Traceback (most recent call last):
File "/mnt/d/AI/projects/digital-human/InfiniteTalk/generate_infinitetalk.py", line 19, in
import wan
File "/mnt/d/AI/projects/digital-human/InfiniteTalk/wan/init.py", line 1, in
from . import configs, distributed, modules
File "/mnt/d/AI/projects/digital-human/InfiniteTalk/wan/modules/init.py", line 1, in
from .attention import flash_attention
File "/mnt/d/AI/projects/digital-human/InfiniteTalk/wan/modules/attention.py", line 5, in
from ..utils.multitalk_utils import RotaryPositionalEmbedding1D, normalize_and_scale, split_token_counts_and_frame_ids
File "/mnt/d/AI/projects/digital-human/InfiniteTalk/wan/utils/init.py", line 7, in
from .vace_processor import VaceVideoProcessor
File "/mnt/d/AI/projects/digital-human/InfiniteTalk/wan/utils/vace_processor.py", line 5, in
import torchvision.transforms.functional as TF
File "/home/sand/miniconda3/envs/ai-infintetalk/lib/python3.10/site-packages/torchvision/init.py", line 10, in
from torchvision import _meta_registrations, datasets, io, models, ops, transforms, utils # usort:skip
File "/home/sand/miniconda3/envs/ai-infintetalk/lib/python3.10/site-packages/torchvision/_meta_registrations.py", line 164, in
def meta_nms(dets, scores, iou_threshold):
File "/home/sand/miniconda3/envs/ai-infintetalk/lib/python3.10/site-packages/torch/library.py", line 828, in register
use_lib._register_fake(op_name, func, _stacklevel=stacklevel + 1)
File "/home/sand/miniconda3/envs/ai-infintetalk/lib/python3.10/site-packages/torch/library.py", line 198, in _register_fake
handle = entry.fake_impl.register(func_to_register, source)
File "/home/sand/miniconda3/envs/ai-infintetalk/lib/python3.10/site-packages/torch/_library/fake_impl.py", line 31, in register
if torch._C._dispatch_has_kernel_for_dispatch_key(self.qualname, "Meta"):
RuntimeError: operator torchvision::nms does not exist
(ai-infintetalk) sand@XiaoQiang10:/mnt/d/AI/projects/digital-human/InfiniteTalk$ python generate_infinitetalk.py
--ckpt_dir weights/Wan2.1-I2V-14B-480P
--wav2vec_dir 'weights/chinese-wav2vec2-base'
--infinitetalk_dir weights/InfiniteTalk/single/infinitetalk.safetensors
--input_json examples/single_example_image.json
--size infinitetalk-480
--sample_steps 40
--mode streaming
--motion_frame 9
--save_file infinitetalk_res

Traceback (most recent call last):
File "/mnt/d/AI/projects/digital-human/InfiniteTalk/generate_infinitetalk.py", line 19, in
import wan
File "/mnt/d/AI/projects/digital-human/InfiniteTalk/wan/init.py", line 1, in
from . import configs, distributed, modules
File "/mnt/d/AI/projects/digital-human/InfiniteTalk/wan/modules/init.py", line 1, in
from .attention import flash_attention
File "/mnt/d/AI/projects/digital-human/InfiniteTalk/wan/modules/attention.py", line 5, in
from ..utils.multitalk_utils import RotaryPositionalEmbedding1D, normalize_and_scale, split_token_counts_and_frame_ids
File "/mnt/d/AI/projects/digital-human/InfiniteTalk/wan/utils/init.py", line 7, in
from .vace_processor import VaceVideoProcessor
File "/mnt/d/AI/projects/digital-human/InfiniteTalk/wan/utils/vace_processor.py", line 5, in
import torchvision.transforms.functional as TF
File "/home/sand/miniconda3/envs/ai-infintetalk/lib/python3.10/site-packages/torchvision/init.py", line 10, in
from torchvision import _meta_registrations, datasets, io, models, ops, transforms, utils # usort:skip
File "/home/sand/miniconda3/envs/ai-infintetalk/lib/python3.10/site-packages/torchvision/_meta_registrations.py", line 164, in
def meta_nms(dets, scores, iou_threshold):
File "/home/sand/miniconda3/envs/ai-infintetalk/lib/python3.10/site-packages/torch/library.py", line 828, in register
use_lib._register_fake(op_name, func, _stacklevel=stacklevel + 1)
File "/home/sand/miniconda3/envs/ai-infintetalk/lib/python3.10/site-packages/torch/library.py", line 198, in _register_fake
handle = entry.fake_impl.register(func_to_register, source)
File "/home/sand/miniconda3/envs/ai-infintetalk/lib/python3.10/site-packages/torch/_library/fake_impl.py", line 31, in register
if torch._C._dispatch_has_kernel_for_dispatch_key(self.qualname, "Meta"):
RuntimeError: operator torchvision::nms does not exist

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions