1. 12 Dec, 2023 1 commit
  2. 06 Dec, 2023 1 commit
    • Lyu Han's avatar
      Report the inference benchmark of models with different size (#794) · ebe90bc9
      Lyu Han authored
      * update test scripts for models with different sizes
      
      * update
      
      * only test after tunning gemm
      
      * chmod +x
      
      * fix typo
      
      * benchmark on a100
      
      * fix typo
      
      * fix typo
      
      * per-token latency percentile in profile_throughput
      
      * fix
      
      * fix
      
      * rename
      
      * make the script accept parameters
      
      * minor fix
      
      * indent
      
      * reformat table
      
      * change to 3000
      
      * minor fix
      ebe90bc9
  3. 04 Dec, 2023 1 commit
  4. 29 Nov, 2023 1 commit
    • Lyu Han's avatar
      Report first-token-latency and token-latency percentiles (#736) · 5c9e1e28
      Lyu Han authored
      * update profile scripts
      
      * add top_p, top_k and temperature as input arguments
      
      * fix input_ids
      
      * update profile_throughput
      
      * update profile_restful_api
      
      * update profile_serving
      
      * update
      
      * update
      
      * add progress bar
      
      * remove TODO comments
      
      * update
      
      * remove useless profile_* argument
      
      * remove log level
      
      * change concurrency default value to 64
      
      * update restful_api.md
      
      * update according to review comments
      
      * fix docstring
      5c9e1e28
  5. 24 Oct, 2023 1 commit
  6. 16 Oct, 2023 1 commit
  7. 18 Sep, 2023 1 commit
  8. 14 Aug, 2023 1 commit
  9. 31 Jul, 2023 1 commit
  10. 23 Jul, 2023 1 commit
    • lvhan028's avatar
      Refactor the chat template of supported models using factory pattern (#144) · 7b470f07
      lvhan028 authored
      * refactor model.py and support baichuan-7b
      
      * remove model_name
      
      * remove hard session_len
      
      * export tokenizer.py to target dir
      
      * remove model_name from client
      
      * remove model_name
      
      * update
      
      * correct throughput equation
      
      * fix session.response
      
      * update serving.md
      
      * update readme
      
      * update according to review comments
      
      * update
      
      * update
      
      * update
      
      * update
      7b470f07
  11. 22 Jul, 2023 1 commit
  12. 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
  13. 06 Jul, 2023 1 commit
  14. 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
  15. 03 Jul, 2023 1 commit
  16. 30 Jun, 2023 2 commits
  17. 25 Jun, 2023 1 commit
    • lvhan028's avatar
      Add profile (#15) · 23c05372
      lvhan028 authored
      * remove constraints on model name
      
      * remove duplicate model converter
      
      * add profile
      
      * get eos and bos from server
      
      * update stop_words
      
      * update sequence_length when the last generated token is eos_id
      
      * fix
      
      * fix
      
      * check-in models
      
      * valicate model_name
      
      * make stop_words as property
      
      * debug profiling
      
      * better stats
      
      * fix assistant reponse
      
      * update profile serving
      
      * update
      
      * update
      23c05372