ColossalAI/examples/inference
Xu Kai 785802e809
[inference] add reference and fix some bugs (#4937)
* add reference and fix some bugs

* update gptq init

---------

Co-authored-by: Xu Kai <xukai16@foxamil.com>
2023-10-20 13:39:34 +08:00
..
serving [Infer] Serving example w/ ray-serve (multiple GPU case) (#4841) 2023-10-02 17:48:38 +08:00
bench_bloom.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
bench_llama.py [Refactor] Integrated some lightllm kernels into token-attention (#4946) 2023-10-19 22:22:47 +08:00
gptq_bloom.py [feature] add gptq for inference (#4754) 2023-09-22 11:02:50 +08:00
gptq_llama.py [inference] add reference and fix some bugs (#4937) 2023-10-20 13:39:34 +08:00
smoothquant_llama.py [inference] Add smmoothquant for llama (#4904) 2023-10-16 11:28:44 +08:00