Meeting size mismatch in torch

#15
by Gofor5 - opened

When I tried to run the workflow, I met
RuntimeError: Error(s) in loading state_dict for AutoencoderKL:
size mismatch for encoder.conv_in.weight: copying a param with shape torch.Size([96, 3, 3, 3, 3]) from checkpoint, the shape in current model is torch.Size([128, 3, 3, 3]).
size mismatch for encoder.conv_in.bias: copying a param with shape torch.Size([96]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for encoder.conv_out.weight: copying a param with shape torch.Size([32, 384, 3, 3, 3]) from checkpoint, the shape in current model is torch.Size([32, 512, 3, 3]).
size mismatch for decoder.conv_in.weight: copying a param with shape torch.Size([384, 16, 3, 3, 3]) from checkpoint, the shape in current model is torch.Size([512, 16, 3, 3]).
size mismatch for decoder.conv_in.bias: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for decoder.conv_out.weight: copying a param with shape torch.Size([3, 96, 3, 3, 3]) from checkpoint, the shape in current model is torch.Size([3, 128, 3, 3]).
size mismatch for quant_conv.weight: copying a param with shape torch.Size([32, 32, 1, 1, 1]) from checkpoint, the shape in current model is torch.Size([32, 32, 1, 1]).
size mismatch for post_quant_conv.weight: copying a param with shape torch.Size([16, 16, 1, 1, 1]) from checkpoint, the shape in current model is torch.Size([16, 16, 1, 1]).
in loading vae, but I am sure that I downloaded the right one . What happened?

I ran into this too. It has nothing to do with nunchaku. You probably downloaded the VAE from https://huggingface.co/Qwen/Qwen-Image/tree/main/vae. You need https://huggingface.co/Comfy-Org/Qwen-Image_ComfyUI/tree/main/split_files/vae instead.

I ran into this too. It has nothing to do with nunchaku. You probably downloaded the VAE from https://huggingface.co/Qwen/Qwen-Image/tree/main/vae. You need https://huggingface.co/Comfy-Org/Qwen-Image_ComfyUI/tree/main/split_files/vae instead.

I have downloaded qwen VAEs from everywhere, this one you linked finally worked, thanks. I wonder why I couldn't happen into this one file in any possible way.

Sign up or log in to comment