github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/inference/modeling/policy
History
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 [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] 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