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/layers
Runyu Lu bcf0181ecd
[Feat] Distrifusion Acceleration Support for Diffusion Inference (#5895)
4 months ago
..
__init__.py
attention.py
baichuan_tp_linear.py [Inference]Lazy Init Support (#5785) 5 months ago
diffusion.py [Feat] Distrifusion Acceleration Support for Diffusion Inference (#5895) 4 months ago
distrifusion.py [Feat] Distrifusion Acceleration Support for Diffusion Inference (#5895) 4 months ago