You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/inference/modeling/policy
Runyu Lu bcf0181ecd
[Feat] Distrifusion Acceleration Support for Diffusion Inference (#5895)
4 months ago
..
__init__.py [Feat] Diffusion Model(PixArtAlpha/StableDiffusion3) Support (#5838) 5 months ago
glide_llama.py [Inference/SpecDec] Support GLIDE Drafter Model (#5455) 8 months ago
nopadding_baichuan.py [Inference]refactor baichuan (#5791) 6 months ago
nopadding_llama.py Pass inference model shard configs for module init 6 months ago
pixart_alpha.py [Feat] Distrifusion Acceleration Support for Diffusion Inference (#5895) 4 months ago
stablediffusion3.py [Feat] Distrifusion Acceleration Support for Diffusion Inference (#5895) 4 months ago