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/tests/test_infer
Zhongkai Zhao 75af66cd81
[Hotfix] Fix model policy matching strategy in ShardFormer (#5064)
1 year ago
..
_utils.py [Hotfix] Fix model policy matching strategy in ShardFormer (#5064) 1 year ago
test_hybrid_bloom.py [inference] update examples and engine (#5073) 1 year ago
test_hybrid_chatglm2.py [inference] update examples and engine (#5073) 1 year ago
test_hybrid_llama.py [inference] update examples and engine (#5073) 1 year ago
test_kvcache_manager.py [inference] Refactor inference architecture (#5057) 1 year ago