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 |
||
---|---|---|
.. | ||
_utils.py | ||
test_bloom_infer.py | ||
test_chatglm2_infer.py | ||
test_infer_engine.py | ||
test_kvcache_manager.py | ||
test_llama_infer.py |