1. 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
  2. 04 Dec, 2023 1 commit
  3. 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
  4. 08 Nov, 2023 1 commit
    • AllentDan's avatar
      fix benchmark serving computation mistake (#630) · 529e56bd
      AllentDan authored
      * fix benchmark serving computation mistake
      
      * fix timestamps computations
      
      * remove speed up
      
      * no mp
      
      * mp seems faster?
      
      * remove
      
      * update
      
      * remove
      
      * fix
      
      * update
      
      * update print log
      
      * typo
      
      * print fist token latency only stream==True
      
      * remove renew_session
      
      * update AsyncEngine
      529e56bd
  5. 01 Nov, 2023 1 commit
    • AllentDan's avatar
      Improve api_server and webui usage (#544) · 373bd013
      AllentDan authored
      * make IPv6 compatible, safe run for coroutine interrupting
      
      * instance_id -> session_id and fix api_client.py
      
      * update doc
      
      * remove useless faq
      
      * safe ip mapping
      
      * update app.py
      
      * WIP completion
      
      * completion
      
      * update doc
      
      * disable interactive mode for /v1/chat/completions
      
      * docstring
      
      * docstring
      
      * refactor gradio
      
      * update gradio
      
      * udpate
      
      * update doc
      
      * rename
      
      * session_id default -1
      
      * missed two files
      
      * add a APIClient
      
      * add chat func for APIClient
      
      * refine
      
      * add concurrent function
      
      * sequence_start, sequence_end --> interactive_mode
      
      * update doc
      
      * comments
      
      * doc
      
      * better text completion
      
      * remove /v1/embeddings
      
      * comments
      
      * deprecate generate and use /v1/interactive/completions
      
      * /v1/interactive/completion -> /v1/chat/interactive
      
      * embeddings
      
      * rename
      
      * remove wrong arg description
      
      * docstring
      
      * fix
      
      * update cli
      
      * update doc
      
      * strict session_len limit condition
      
      * pass model args to api_server
      373bd013
  6. 24 Oct, 2023 1 commit
  7. 16 Oct, 2023 1 commit
  8. 11 Oct, 2023 1 commit
  9. 26 Sep, 2023 1 commit
  10. 18 Sep, 2023 1 commit
  11. 04 Sep, 2023 1 commit
  12. 24 Aug, 2023 1 commit
    • WRH's avatar
      [Feature] Support decode with DP in pytorch (#193) · 81f29837
      WRH authored
      * support decode
      
      * unit test and benckmark and improve
      
      * remove some drafts
      
      * enable numerical test
      
      * minor
      
      * add some benchmark data
      
      * add more output
      
      * update interface
      
      * remove debugs
      
      * format
      
      * update docstring
      
      * remove print and add benchmark results
      
      * use logits & add main
      
      * fix rb
      
      * dump large
      
      * update test
      
      * update test decode
      
      * add decimal
      81f29837
  13. 22 Aug, 2023 1 commit
    • AllentDan's avatar
      Add Restful API (#223) · d5c10e7a
      AllentDan authored
      * add restful api
      
      * refine
      
      * add simple doc
      
      * lint
      
      * add uvicorn requirement
      
      * more args
      
      * add llama2
      
      * docstring
      
      * update doc
      
      * save
      
      * refine
      
      * lint
      
      * better decode
      
      * add v1/embedding
      
      * add GenerateRequest
      
      * add llama2 chat template
      
      * correct profiling
      
      * update documents
      
      * add length judge
      
      * add faq
      
      * update doc and rename req_que to req_queue
      
      * fix md link, use get_logger, fix sequence_end bug
      
      * use another doc link for go to avoid lint error
      
      * add api_client.py
      
      * update doc
      
      * update doc
      
      * update function interface
      
      * update FAQ
      
      * resolve comments
      d5c10e7a
  14. 16 Aug, 2023 1 commit
  15. 14 Aug, 2023 1 commit
  16. 07 Aug, 2023 1 commit
  17. 31 Jul, 2023 2 commits
  18. 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
  19. 22 Jul, 2023 1 commit
  20. 19 Jul, 2023 1 commit
  21. 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
  22. 06 Jul, 2023 1 commit
  23. 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
  24. 03 Jul, 2023 1 commit
  25. 30 Jun, 2023 2 commits
  26. 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