ColossalAI/colossalai/inference/modeling/models
Runyu Lu cba20525a8
[Feat] Diffusion Model(PixArtAlpha/StableDiffusion3) Support (#5838)
* Diffusion Model Inference support

* Stable Diffusion 3 Support

* pixartalpha support
2024-07-08 16:02:07 +08:00
..
__init__.py fix bugs in request_handler 2024-01-11 13:39:56 +00:00
diffusion.py [Feat] Diffusion Model(PixArtAlpha/StableDiffusion3) Support (#5838) 2024-07-08 16:02:07 +08:00
glide_llama.py [Fix] Fix spec-dec Glide LlamaModel for compatibility with transformers (#5837) 2024-06-19 15:37:53 +08:00
nopadding_baichuan.py [Inference]Lazy Init Support (#5785) 2024-06-27 18:02:15 +08:00
nopadding_llama.py [Inference]Lazy Init Support (#5785) 2024-06-27 18:02:15 +08:00
pixart_alpha.py [Feat] Diffusion Model(PixArtAlpha/StableDiffusion3) Support (#5838) 2024-07-08 16:02:07 +08:00
stablediffusion3.py [Feat] Diffusion Model(PixArtAlpha/StableDiffusion3) Support (#5838) 2024-07-08 16:02:07 +08:00