ColossalAI/colossalai/inference/modeling/policy
Runyu Lu bcf0181ecd
[Feat] Distrifusion Acceleration Support for Diffusion Inference (#5895)
* Distrifusion Support source

* comp comm overlap optimization

* sd3 benchmark

* pixart distrifusion bug fix

* sd3 bug fix and benchmark

* generation bug fix

* naming fix

* add docstring, fix counter and shape error

* add reference

* readme and requirement
2024-07-30 10:43:26 +08:00
..
__init__.py [Feat] Diffusion Model(PixArtAlpha/StableDiffusion3) Support (#5838) 2024-07-08 16:02:07 +08:00
glide_llama.py [Inference/SpecDec] Support GLIDE Drafter Model (#5455) 2024-04-10 11:07:52 +08:00
nopadding_baichuan.py [Inference]refactor baichuan (#5791) 2024-06-11 10:52:01 +08:00
nopadding_llama.py Pass inference model shard configs for module init 2024-06-07 08:33:52 +00:00
pixart_alpha.py [Feat] Distrifusion Acceleration Support for Diffusion Inference (#5895) 2024-07-30 10:43:26 +08:00
stablediffusion3.py [Feat] Distrifusion Acceleration Support for Diffusion Inference (#5895) 2024-07-30 10:43:26 +08:00