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

Update train.py to use image_size for validation image dimensions #74

Closed congruency closed 2 months ago

congruency commented 2 months ago

Replaced fixed 1024 with image_size from config.image_size