1. 13 Dec, 2023 1 commit
    • AllentDan's avatar
      Add api.py (#805) · 5c9aa51a
      AllentDan authored
      * add api.py
      
      * update serve function
      
      * add model_name arg and provide examples
      
      * docstring
      
      * remove service_available
      
      * type hint
      5c9aa51a
  2. 25 Oct, 2023 1 commit
    • RunningLeon's avatar
      Add more user-friendly CLI (#541) · 169d5169
      RunningLeon authored
      * add
      
      * import fire in main
      
      * wrap to speed up fire cli
      
      * update
      
      * update docs
      
      * update docs
      
      * fix
      
      * resolve commennts
      
      * resolve confict and add test for cli
      169d5169
  3. 19 Oct, 2023 1 commit
  4. 16 Oct, 2023 1 commit
  5. 26 Sep, 2023 1 commit
    • AllentDan's avatar
      expose stop words and filter eoa (#352) · 327deaee
      AllentDan authored
      * expose stop words
      
      * support string
      
      * fix
      
      * remove eoa from chatbot
      
      * remove eoa of turbomind
      
      * fix ut
      
      * suffix wheel and fix InternLM no system bug
      327deaee
  6. 20 Sep, 2023 1 commit
    • Lyu Han's avatar
      Support InternLM 20B (#440) · df7955de
      Lyu Han authored
      
      
      * better profiler
      
      * wait for releasing mem
      
      * remove fire
      
      * remove support for multiple model benchmark
      
      * comments
      
      * support actual seqlen
      
      * change chat template
      
      * update
      
      * fix ut
      
      * int->size_t
      
      * output more details
      
      * correct tp
      
      * rollback
      
      * update
      
      * update readme
      
      * add 'internlm-chat' as the default tag for internlm chat models
      
      * rollback tokenizer
      
      ---------
      Co-authored-by: default avatarAllentDan <AllentDan@yeah.net>
      Co-authored-by: default avatargrimoire <yaoqian@pjlab.org.cn>
      df7955de
  7. 18 Sep, 2023 1 commit
  8. 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
  9. 29 Aug, 2023 1 commit
  10. 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
  11. 17 Aug, 2023 1 commit
    • Chen Xin's avatar
      Support windows platform (#209) · 4c9959f6
      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
      4c9959f6
  12. 07 Aug, 2023 1 commit
    • WRH's avatar
      [Refactor] Support multi-session chat (#178) · 4bd0b487
      WRH authored
      * add some dist utils
      
      * add model utils
      
      * add termio and basicstreamer
      
      * typo
      
      * fix world size
      
      * refactor chat and tested llama1
      
      * add internlm adapter and support stoping criteria
      
      * concat with id for internlm
      
      * update docstring
      
      * update and support llama2
      
      * typo
      
      * move docs to docs
      
      * update docstring of session manager
      
      * update docstring
      
      * update docs
      
      * fix accel none in model
      
      * fix and add test for tensor broadcast
      
      * fix session using typing to check type
      
      * add docstrings and comprehensive condition test
      
      * unit test for dist
      
      * fix session
      
      * split unittests of utils
      
      * typo
      
      * update control flow of accel
      
      * move test model
      
      * remove main in unittest
      
      * remove some log
      
      * remove some comments
      4bd0b487
  13. 03 Aug, 2023 1 commit
  14. 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
  15. 17 Jul, 2023 1 commit
  16. 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
  17. 01 Jul, 2023 3 commits
  18. 20 Jun, 2023 1 commit