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
Xu Kai c6295c3381
[Refactor] remove useless inference code (#5022)
1 year ago
..
_utils.py [misc] update pre-commit and run all files (#4752) 1 year ago
test_hybrid_bloom.py [Refactor] remove useless inference code (#5022) 1 year ago
test_hybrid_chatglm2.py [Inference Refactor] Merge chatglm2 with pp and tp (#5023) 1 year ago
test_hybrid_llama.py [Refactor] remove useless inference code (#5022) 1 year ago
test_kvcache_manager.py [Refactor] remove useless inference code (#5022) 1 year ago