mirror of https://github.com/hpcaitech/ColossalAI
![]() * add infer_struct and infer_config * update codes * change InferConfig * Add hf_model_config to the engine * rm _get_hf_model_config * update codes * made adjustments according to the feedback from the reviewer. * update codes * add ci test for config and struct * Add the logic of the inference engine * update engine and test * Recover cache_manager.py * add logger * fix conflict * update codes * update codes * update model and tokenizer * fix add the logic about shardformer * change kvcache_manager docstring * add policy * fix ci bug in test_kvcache_manager.py * remove codes related o tokenizer and move model_policy * fix code style * add ordered_set to requirements-infer.txt * Delete extra empty lines * add ordered_set to requirements-test.txt |
||
---|---|---|
.. | ||
kit | ||
test_analyzer | ||
test_auto_parallel | ||
test_autochunk | ||
test_booster | ||
test_checkpoint_io | ||
test_cluster | ||
test_config | ||
test_device | ||
test_fx | ||
test_gptq | ||
test_infer | ||
test_infer_ops/triton | ||
test_lazy | ||
test_legacy | ||
test_moe | ||
test_optimizer | ||
test_pipeline | ||
test_shardformer | ||
test_smoothquant | ||
test_tensor | ||
test_utils | ||
test_zero | ||
__init__.py |