1. 01 Sep, 2023 1 commit
  2. 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
  3. 21 Aug, 2023 1 commit
    • RunningLeon's avatar
      add readthedocs (#208) · c238f1cd
      RunningLeon authored
      * add readthedocs configs
      
      * update readme
      
      * fix link
      
      * update
      
      * remove turbomind in api
      
      * update
      
      * fix comment and remove api
      c238f1cd
  4. 14 Aug, 2023 1 commit
  5. 11 Aug, 2023 1 commit
    • pppppM's avatar
      [Feature] Support AWQ (#108) · d3dbe179
      pppppM authored
      * support kv cache offload
      
      * add dataloader docstring
      
      * complete gitignore
      
      * refactor collect mod fn
      
      * add calibration
      
      * fix lint
      
      * add observers and quantizers
      
      * fix lints
      
      * add global available mixin
      
      * fix lints
      
      * split batch inference
      
      * support smoothquant and awq
      
      * update export kv scales
      
      * fix lints
      
      * fix some bugs
      
      * update weight only usage
      
      * update usage
      
      * auto mapping and support smooth internlm
      
      * trust remote code
      
      * fix num head key error
      
      * fix bias error
      
      * align shape and pack order with llm-awq
      
      * modified according to LZHgrla's comments.
      
      * update gitignore
      
      * fix kv qparams export error
      
      * update usage
      
      * decouple calibrate and awq
      
      * update docstrings
      
      * update api name
      
      * update readme
      
      * update readme
      
      * update readme
      
      * update readme
      
      * update kv_qparams and readme
      
      * fix typos
      d3dbe179
  6. 27 Jul, 2023 1 commit
    • Chen Xin's avatar
      Add manylinux builder (#164) · b9004712
      Chen Xin authored
      
      
      * update builder
      
      * remove root permission
      
      * update readme
      
      * update setup.py
      
      * add install cuda 12.1 script
      
      * use generate.sh
      
      * add nccl to install_requires
      
      * update README.md
      
      * fix lint
      
      * update setup.py
      
      ---------
      Co-authored-by: default avatarchenxin <chenxin@pjlab.org.cn>
      b9004712
  7. 06 Jul, 2023 3 commits
    • WRH's avatar
      [Feature] Add a torch client (#19) · 009075d8
      WRH authored
      * draft torch client
      
      * deal with space of tokenizer
      
      * support tensor parallel
      
      * fix
      
      * fix
      
      * move folder
      
      * move instruction to readme
      
      * move to torch/
      
      * rename client to chat
      
      * very bad response
      
      * stash
      
      * rename streamer
      
      * support internlm
      
      * change default args
      
      * remove test
      
      * improve instructions
      
      * remove module docstring
      
      * decrease header level of torch model
      009075d8
    • tpoisonooo's avatar
      Update .gitignore (#70) · b2393467
      tpoisonooo authored
      * docs(README): fix script
      b2393467
    • tpoisonooo's avatar
      fix(project): interlm run error (#69) · 22d403f5
      tpoisonooo authored
      22d403f5
  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
  9. 04 Jul, 2023 1 commit
  10. 20 Jun, 2023 2 commits
  11. 18 Jun, 2023 1 commit