- 11 Dec, 2023 1 commit
-
-
Li Zhang authored
* simplify block manager * fix lint
-
- 07 Dec, 2023 1 commit
-
-
Li Zhang authored
-
- 06 Dec, 2023 1 commit
-
-
Lyu Han authored
-
- 04 Dec, 2023 1 commit
-
-
Li Zhang authored
* Unify prefill and decode passes * dynamic split-fuse * refactor * correct input count calculation * remove unused * lint * lint * fix msvc build * fix msvc build * fix msvc build * fix msvc build * fix msvc build * fix msvc build * fix msvc build * fix msvc build * fix msvc build
-
- 02 Dec, 2023 1 commit
-
-
Li Zhang authored
-
- 29 Nov, 2023 2 commits
-
-
tpoisonooo authored
* feat(build): enable ninja and lld * fix(.github): add ninja installation * fix(CI): remove dimsize=256 * fix(CI): add option for generate.sh * fix(docs): update
-
q.yao authored
* fix * fix lint
-
- 28 Nov, 2023 1 commit
-
-
q.yao authored
-
- 23 Nov, 2023 2 commits
- 22 Nov, 2023 1 commit
-
-
Chen Xin authored
* turbomind support export model params * fix overflow * support turbomind.from_pretrained * fix tp * support AutoModel * support load kv qparams * update auto_awq * udpate docstring * export lmdeploy version * update doc * remove download_hf_repo * LmdeployForCausalLM -> LmdeployForCausalLM * refactor turbomind.py * update comment * add bfloat16 convert back * support gradio run_locl load hf * support resuful api server load hf * add docs * support loading previous quantized model * adapt pr 690 * udpate docs * not export turbomind config when quantize a model * check model_name when can not get it from config.json * update readme * remove model_name in auto_awq * update * update * udpate * fix build * absolute import
-
- 20 Nov, 2023 1 commit
-
-
Li Zhang authored
* tmp * update * update * optimize for throughput * update * fix eos * clean up * fix serving * fix indexed copy * minor * minor --------- Co-authored-by:lvhan028 <lvhan_028@163.com>
-
- 14 Nov, 2023 1 commit
-
-
Li Zhang authored
* fix init of finished buf * fix `finished_count`
-
- 10 Nov, 2023 1 commit
-
-
Li Zhang authored
* refresh decoder attention kernel * block-level kv cache * `BlockManager` & `SequenceManager` * update * update * update * update * rename * GQA support * fix context length * GQA dispatch * kv8 * tune * async stream cb * nvtx * config parsing * debug * optimize output cost * split-k decoding * minor * truncate `session_len` by available blocks * minor * license * fix * dispatch `cp.async` * fix linking * fix * fix deadlock * guard input length * correct start offset * fix prefill chunking * fix `cache_block_seq_len` param passing * fix `block_size` fmtstr * fix output tokens * fix batch resizing * fix masking of finished sequences * add debug util * free unused block early * add ntk scaling and logn scaling * cmake flags * fix typo * w4a16 for sm75 * fix msvc build * fix msvc build * fix block verification * fix msvc build * use `std::shuffle` * fix lint * fix lint * fix lint * clear incoming buffer * clear finished requests * fix batch initialization * fix typo * fix typo * fix comparison
-
- 11 Oct, 2023 1 commit
-
-
akhoroshev authored
-
- 09 Oct, 2023 1 commit
-
-
Lyu Han authored
* change shared_instances_ from weakptr to sharedptr * update
-
- 26 Sep, 2023 3 commits
-
-
Lyu Han authored
* Fix memory leak * modern c++
-
akhoroshev authored
-
akhoroshev authored
* cuda allocator fix * graceful termination * lint and compilation fix
-
- 18 Sep, 2023 2 commits
- 14 Sep, 2023 1 commit
-
-
Chen Xin authored
-
- 11 Sep, 2023 1 commit
-
-
Lyu Han authored
* tmp * add demo for codellama inference * update * update * update * update codellama.md * export rope_theta * update * update doc * fix client.py * define SamplingParam * rollback 'end' * rotary_emb_base to rotary_embedding_base * change to baichuan2-7b
-
- 07 Sep, 2023 1 commit
-
-
Lyu Han authored
* Fix crash when context window size is large by setting max dynamic smem size * fix linting
-
- 01 Sep, 2023 1 commit
-
-
Chen Xin authored
* pack llama_gemm * update CMakeLists.txt * remove candidate * update MANIFEST.in
-
- 29 Aug, 2023 1 commit
-
-
q.yao authored
* first * fix causal mask * disable flash attention2 on sm70 * fix 2 * update readme * clang-format * disable ft2 on windows * fix lint * fix build * fix build * fix long kv seq * fix lint * sync copy output --------- Co-authored-by:
grimoire <yaoqian@pjlab.org.cn> Co-authored-by:
irexyc <irexyc@gmail.com>
-
- 24 Aug, 2023 1 commit
-
-
Lyu Han authored
* Pad tok_embedding and output weights to make their shape divisible by TP * update * update * update * update * update llamaBatch
-
- 22 Aug, 2023 1 commit
-
-
Li Zhang authored
* disable cache hint for CUDA < 11.4 * fix lint * fix lint * fix cuda-11.3 build
-
- 18 Aug, 2023 1 commit
-
-
Li Zhang authored
* qwen support * dynamic ntk & logn attn * fix ntk & add chat template * fix ntk scaling & stop words * fix lint * add tiktoken to requirements.txt * fix tokenizer, set model format automatically * update model.py * update readme * fix lint
-
- 17 Aug, 2023 2 commits
-
-
Li Zhang authored
-
Chen Xin authored
* __PRETTY_FUNCTION__ * CASE_K * uint * remove not * HALF_FLT_MAX * struct init * port utils * better build pthread-win32 * port kernels * port utils/gemm_test * hide windows header * port models * port examples && triton_backend && unittests * update build readme * fix lint * fix lint * fix lint * fix lint * fix lint * fix build * fix build * cmake version * fix typos * update ci * port kernels/gemm_s_f16 * update ci * fix ci * use cudaStreamSynchronize instead of volatile check * remove gettimeofday * remove pthread-win32 * remove dirent.h * update pre-commit * update * remove todo * fix include * fix build * fix build * fix build ci * fix github action trigger * update README * fix linux-build ci * remove windows folder * fix lint * update readme
-
- 15 Aug, 2023 1 commit
-
-
Chen Xin authored
-
- 14 Aug, 2023 2 commits
-
-
tpoisonooo authored
* feat(quantization): kv cache use asymmetric
-
Li Zhang authored
* add w4a16 * fix `deploy.py` * add doc * add w4a16 kernels * fuse w1/w3 & bugfixes * fix typo * python * guard sm75/80 features * add missing header * refactor * qkvo bias * update cost model * fix lint * update `deploy.py`
-
- 03 Aug, 2023 1 commit
-
-
lvhan028 authored
* fix build tests failure * move src test cases to tests/csrc
-
- 31 Jul, 2023 2 commits
- 27 Jul, 2023 1 commit
-
-
Chen Xin authored
* update builder * remove root permission * update readme * update setup.py * add install cuda 12.1 script * use generate.sh * add nccl to install_requires * update README.md * fix lint * update setup.py --------- Co-authored-by:chenxin <chenxin@pjlab.org.cn>
-
- 25 Jul, 2023 1 commit
-
-
q.yao authored
Co-authored-by:grimoire <yaoqian@pjlab.org.cn>
-
- 24 Jul, 2023 1 commit
-
-
Li Zhang authored
* decode only forward pass * fix lint * batch embedding
-