Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

denoising_unet权重问题 #102

Open
wzr0108 opened this issue Dec 25, 2024 · 1 comment
Open

denoising_unet权重问题 #102

wzr0108 opened this issue Dec 25, 2024 · 1 comment
Labels
duplicate This issue or pull request already exists

Comments

@wzr0108
Copy link

wzr0108 commented Dec 25, 2024

我打印了加载denoising_unet权重的信息

    missing_keys, unexpected_keys = denoising_unet.load_state_dict(
        torch.load(config.denoising_unet_path, map_location="cpu"), strict=False
    )
    print(f"Loading {config.denoising_unet_path}")
    print(f"missing_keys: {missing_keys}")
    print(f"unexpected_keys: {unexpected_keys}")

输出了很多missing_keys

Loading ./pretrained_weights/denoising_unet.pth
missing_keys: ['down_blocks.0.motion_modules.0.temporal_transformer.norm.weight', 'down_blocks.0.motion_modules.0.temporal_transformer.norm.bias', 'down_blocks.0.motion_modules.0.temporal_transformer.proj_in.weight', 'down_blocks.0.motion_modules.0.temporal_transformer.proj_in.bias', 

这是正常的吗

@mengrang
Copy link
Collaborator

按说明配置就没关系

@mengrang mengrang added done invalid This doesn't seem right duplicate This issue or pull request already exists and removed done invalid This doesn't seem right labels Dec 25, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
duplicate This issue or pull request already exists
Projects
None yet
Development

No branches or pull requests

2 participants