Skip to content

ImportError: cannot import name 'consume_prefix_in_state_dict_if_present' from 'torch.nn.modules.utils' #22

Description

@zxj329

加载hubert-soft模型的时候,hubert = torch.hub.load("bshall/hubert:main", f"hubert_soft").to(device).eval()报了这个错误,我的torch版本是1.8.0+cu111,请问下需要哪个torch版本。如果使用torch2.x以上的版本,你代码中的stft相关的会报错。很矛盾的事情

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions