File size: 699 Bytes
8d58dd2 | 1 2 3 4 5 6 7 8 9 | ---
pipeline_tag: unconditional-image-generation
---
# FiTv2: Scalable and Improved Flexible Vision Transformer for Diffusion Model
This repository contains the FiTv2 model described in [FiTv2: Scalable and Improved Flexible Vision Transformer for Diffusion Model](https://huggingface.co/papers/2410.13925). FiTv2 is a flexible vision transformer for diffusion models that generates images at unrestricted resolutions and aspect ratios. It builds on FiT with innovations such as Query-Key vector normalization, AdaLN-LoRA, a rectified flow scheduler, and a Logit-Normal sampler.
The official code and sampling scripts are available at [https://github.com/whlzy/FiT](https://github.com/whlzy/FiT). |