ZHO-ZHO-ZHO/ComfyUI-PixArt-alpha-Diffusers

Cannot load <class 'diffusers.models.transformers.transformer_2d.Transformer2DModel'>

vivek-kumar-poddar opened this issue · 0 comments

I am getting this error:

!!! Exception during processing!!! Cannot load <class 'diffusers.models.transformers.transformer_2d.Transformer2DModel'> from C:\Users\VKP.cache\huggingface\hub\models--PixArt-alpha--PixArt-Sigma-XL-2-1024-MS\snapshots\937eb4b6ca35a159c56494998176988a466f83b5\transformer because the following keys are missing:
adaln_single.emb.resolution_embedder.linear_1.weight, adaln_single.emb.aspect_ratio_embedder.linear_1.bias, adaln_single.emb.aspect_ratio_embedder.linear_2.weight, adaln_single.emb.resolution_embedder.linear_2.bias, adaln_single.emb.resolution_embedder.linear_2.weight, adaln_single.emb.aspect_ratio_embedder.linear_1.weight, adaln_single.emb.resolution_embedder.linear_1.bias, adaln_single.emb.aspect_ratio_embedder.linear_2.bias.

Please make sure to pass low_cpu_mem_usage=False and device_map=None if you want to randomly initialize those weights or else make sure your checkpoint file is correct.

My system config is:

17-14700k
4070-ti
64gb ddr4 3600 ram
Nvme ssd