1. 24 Mar, 2024 1 commit
  2. 22 Mar, 2024 1 commit
  3. 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
  4. 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
  5. 19 Oct, 2023 1 commit
  6. 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
  7. 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
  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. 05 Jul, 2023 1 commit
    • 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