1. 16 Oct, 2023 1 commit
  2. 11 Sep, 2023 1 commit
    • Lyu Han's avatar
      Support codellama (#359) · 65c662f9
      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
      65c662f9
  3. 01 Sep, 2023 1 commit
    • AllentDan's avatar
      Decode generated token_ids incrementally (#309) · 9bfe03c6
      AllentDan authored
      * add incremental decoding for turbomind
      
      * update TIS
      
      * fix triton post processing
      
      * update doc
      
      * fix typo
      
      * SentencePieceTokenizer incremental decode, add qwen message prompt
      
      * docstring
      
      * update bot
      9bfe03c6
  4. 18 Aug, 2023 1 commit
  5. 22 Jul, 2023 1 commit
  6. 18 Jul, 2023 1 commit
    • q.yao's avatar
      Tensor Parallel python api (#82) · 7cbfe2ea
      q.yao authored
      * wip
      
      * profile disable tp
      
      * fix profile
      
      * lint
      
      * fix dlpack
      
      * remove comment
      
      * add tp flag
      
      * add session len check
      
      * add eos
      
      * remove tp and session len inputs
      
      * warp tokenizer
      
      * multithread load weight
      
      * update profile
      
      * refactor tokenizer
      
      * remove pre/post process
      
      * remove mpi4py requirement
      
      * remove
      
      * remove bind
      
      * remove mpi requirement
      
      * check backend_tokenizer
      7cbfe2ea
  7. 12 Jul, 2023 1 commit
  8. 05 Jul, 2023 2 commits
    • pppppM's avatar
      [Feature] Stats Quantization Parameters for KV Cache (#45) · 3fff964d
      pppppM authored
      * add cal qparams
      
      * support offload inference
      
      * add collect funtions (mod,weight)
      
      * stats kv scales
      
      * update init
      
      * add user guide
      
      * fix hints
      
      * fix comments & support turbomind format
      
      * update user guide
      
      * fix slice kv cache error & support pileval dataset (used in llm-awq)
      
      * fix wrong num heads slice
      
      * update default dataset
      
      * fix conflict
      
      * fix hints
      
      * fix hints
      
      * add gitignore
      3fff964d
    • q.yao's avatar
      Python ffi (#34) · 4fd6e710
      q.yao authored
      
      
      * wip
      
      * wip
      
      * example finish
      
      * fix include and namespace
      
      * wtf
      
      * install lib
      
      * batchize
      
      * update cmake install
      
      * multithread
      
      * fix comment
      
      * fix
      
      * add mmengine
      
      * bind llamamodel
      
      ---------
      Co-authored-by: default avatargrimoire <yaoqian@pjlab.org.cn>
      4fd6e710