mirror of https://github.com/hpcaitech/ColossalAI
ce7ade3882
* add chatglm2 * add * gather needed kernels * fix some bugs * finish context forward * finish context stage * fix * add * pause * add * fix bugs * finish chatglm * fix bug * change some logic * fix bugs * change some logics * add * add * add * fix * fix tests * fix |
||
---|---|---|
.. | ||
__init__.py | ||
albert.py | ||
bert.py | ||
blip2.py | ||
bloom.py | ||
chatglm2.py | ||
gpt.py | ||
llama.py | ||
opt.py | ||
sam.py | ||
t5.py | ||
vit.py | ||
whisper.py |