Major overhaul / better native integration - #92

Draft

city96 wants to merge 15 commits into

mainfrom

Commits on Dec 10, 2024

PixArt initial rewrite
city96
committed
3f844ca
Remove HyDiT - supported in main
city96
committed
45423d9
Use native ops for PixArt
city96
committed
d5a47fa

Commits on Dec 11, 2024

Unused
city96
committed
6a88140
Consolidate loader logic
city96
committed
06a9368
Sana loader logic
city96
committed
5505ab4
Use native ops for sana
city96
committed
7e19ac5
Faster model initialization
city96
committed
2beff5a
Patch size logic from common dit
city96
committed
d1f7516
Make mlp args match the one from timm
city96
committed
ca2bc8b
Remove timm as a dependency
city96
committed
ee982c5
Add switch for new xformers
city96
committed
4a11543

Commits on Dec 12, 2024

Add fast rgb preview to Sana
city96
committed
7b3f2ce
Rename/move
city96
committed
c52d827

Commits on Dec 13, 2024

Gemma first implementation
city96
committed
0f567fc