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 68 forks source link

Store model on CPU during convert_pixart_to_diffusers.py process #66

Closed congruency closed 2 months ago

congruency commented 2 months ago

Loads the torch model into CPU memory for systems with low VRAM available.