Modify inference.py to allow any image_size

PixArt-alpha / PixArt-sigma

PixArt-Σ: Weak-to-Strong Training of Diffusion Transformer for 4K Text-to-Image Generation

GNU Affero General Public License v3.0

1.44k stars 67 forks source link

Closed congruency closed 1 week ago

congruency commented 1 month ago

Adds support for different sizes than the native training resolution to scripts/inference.py #99

Calculates pe_interpolation instead of lookup table.
Falls back to 1024 ratios when none found for the specified image_size rather than throwing an exception.

768x768 A woman smiling

1280x1280 A woman smiling