search

PixArt-alpha / PixArt-sigma

PixArt-Σ: Weak-to-Strong Training of Diffusion Transformer for 4K Text-to-Image Generation
https://pixart-alpha.github.io/PixArt-sigma-project/
GNU Affero General Public License v3.0
1.44k stars 67 forks source link

Weight Issues? Trying to install in comfy UI #129

Closed karl0ss closed 1 day ago

karl0ss commented 2 days ago

I have followed a guide on a youttube video to get this working in ComfyUI and everything looked to work ok, but when I try and run the suggested workflow I am getting the following message 

RuntimeError: Error(s) in loading state_dict for PixArtMS:
    size mismatch for y_embedder.y_embedding: copying a param with shape torch.Size([300, 4096]) from checkpoint, the shape in current model is torch.Size([120, 4096]).

Wondering if someone can give me some tips?

Thanks

Karl

karl0ss commented 1 day ago

Looks li was was using the wrong Resolution Select against the wrong Checkpoint