1. 04 Oct, 2023 1 commit
  2. 02 Oct, 2023 2 commits
    • Yuanheng Zhao's avatar
      [Infer] Serving example w/ ray-serve (multiple GPU case) (#4841) · 573f2705
      Yuanheng Zhao authored
      * fix imports
      
      * add ray-serve with Colossal-Infer tp
      
      * trivial: send requests script
      
      * add README
      
      * fix worker port
      
      * fix readme
      
      * use app builder and autoscaling
      
      * trivial: input args
      
      * clean code; revise readme
      
      * testci (skip example test)
      
      * use auto model/tokenizer
      
      * revert imports fix (fixed in other PRs)
      573f2705
    • Yuanheng Zhao's avatar
      [Infer] Colossal-Inference serving example w/ TorchServe (single GPU case) (#4771) · 3a74eb4b
      Yuanheng Zhao authored
      * add Colossal-Inference serving example w/ TorchServe
      
      * add dockerfile
      
      * fix dockerfile
      
      * fix dockerfile: fix commit hash, install curl
      
      * refactor file structure
      
      * revise readme
      
      * trivial
      
      * trivial: dockerfile format
      
      * clean dir; revise readme
      
      * fix comments: fix imports and configs
      
      * fix formats
      
      * remove unused requirements
      3a74eb4b
  3. 28 Sep, 2023 2 commits
  4. 27 Sep, 2023 8 commits
  5. 26 Sep, 2023 8 commits
  6. 25 Sep, 2023 2 commits
  7. 24 Sep, 2023 2 commits
  8. 22 Sep, 2023 4 commits
    • Hongxin Liu's avatar
      [release] update version (#4775) · 4146f1c0
      Hongxin Liu authored
      * [release] update version
      
      * [doc] revert versions
      4146f1c0
    • Jianghai's avatar
      [inference] chatglm2 infer demo (#4724) · ce7ade38
      Jianghai authored
      * add chatglm2
      
      * add
      
      * gather needed kernels
      
      * fix some bugs
      
      * finish context forward
      
      * finish context stage
      
      * fix
      
      * add
      
      * pause
      
      * add
      
      * fix bugs
      
      * finish chatglm
      
      * fix bug
      
      * change some logic
      
      * fix bugs
      
      * change some logics
      
      * add
      
      * add
      
      * add
      
      * fix
      
      * fix tests
      
      * fix
      ce7ade38
    • Xu Kai's avatar
      [feature] add gptq for inference (#4754) · 946ab56c
      Xu Kai authored
      * [gptq] add gptq kernel (#4416)
      
      * add gptq
      
      * refactor code
      
      * fix tests
      
      * replace auto-gptq
      
      * rname inferance/quant
      
      * refactor test
      
      * add auto-gptq as an option
      
      * reset requirements
      
      * change assert and check auto-gptq
      
      * add import warnings
      
      * change test flash attn version
      
      * remove example
      
      * change requirements of flash_attn
      
      * modify tests
      
      * [skip ci] change requirements-test
      
      * [gptq] faster gptq cuda kernel (#4494)
      
      * [skip ci] add cuda kernels
      
      * add license
      
      * [skip ci] fix max_input_len
      
      * format files & change test size
      
      * [skip ci]
      
      * [gptq] add gptq tensor parallel (#4538)
      
      * add gptq tensor parallel
      
      * add gptq tp
      
      * delete print
      
      * add test gptq check
      
      * add test auto gptq check
      
      * [gptq] combine gptq and kv cache manager (#4706)
      
      * combine gptq and kv cache manager
      
      * add init bits
      
      * delete useless code
      
      * add model path
      
      * delete usless print and update test
      
      * delete usless import
      
      * move option gptq to shard config
      
      * change replace linear to shardformer
      
      * update bloom policy
      
      * delete useless code
      
      * fix import bug and delete uselss code
      
      * change colossalai/gptq to colossalai/quant/gptq
      
      * update import linear for tests
      
      * delete useless code and mv gptq_kernel to kernel directory
      
      * fix triton kernel
      
      * add triton import
      946ab56c
    • littsk's avatar
      [bug] Fix the version check bug in colossalai run when generating the cmd. (#4713) · 1e0e0808
      littsk authored
      * Fix the version check bug in colossalai run when generating the cmd.
      
      * polish code
      1e0e0808
  9. 21 Sep, 2023 5 commits
  10. 20 Sep, 2023 4 commits
    • Baizhou Zhang's avatar
      [shardformer] fix master param sync for hybrid plugin/rewrite unwrapping logic (#4758) · c0a03370
      Baizhou Zhang authored
      * fix master param sync for hybrid plugin
      
      * rewrite unwrap for ddp/fsdp
      
      * rewrite unwrap for zero/gemini
      
      * rewrite unwrap for hybrid plugin
      
      * fix geemini unwrap
      
      * fix bugs
      c0a03370
    • Wenhao Chen's avatar
      [chat]: update rm, add wandb and fix bugs (#4471) · 7b9b8644
      Wenhao Chen authored
      
      
      * feat: modify forward fn of critic and reward model
      
      * feat: modify calc_action_log_probs
      
      * to: add wandb in sft and rm trainer
      
      * feat: update train_sft
      
      * feat: update train_rm
      
      * style: modify type annotation and add warning
      
      * feat: pass tokenizer to ppo trainer
      
      * to: modify trainer base and maker base
      
      * feat: add wandb in ppo trainer
      
      * feat: pass tokenizer to generate
      
      * test: update generate fn tests
      
      * test: update train tests
      
      * fix: remove action_mask
      
      * feat: remove unused code
      
      * fix: fix wrong ignore_index
      
      * fix: fix mock tokenizer
      
      * chore: update requirements
      
      * revert: modify make_experience
      
      * fix: fix inference
      
      * fix: add padding side
      
      * style: modify _on_learn_batch_end
      
      * test: use mock tokenizer
      
      * fix: use bf16 to avoid overflow
      
      * fix: fix workflow
      
      * [chat] fix gemini strategy
      
      * [chat] fix
      
      * sync: update colossalai strategy
      
      * fix: fix args and model dtype
      
      * fix: fix checkpoint test
      
      * fix: fix requirements
      
      * fix: fix missing import and wrong arg
      
      * fix: temporarily skip gemini test in stage 3
      
      * style: apply pre-commit
      
      * fix: temporarily skip gemini test in stage 1&2
      
      ---------
      Co-authored-by: default avatarMingyan Jiang <1829166702@qq.com>
      7b9b8644
    • ppt0011's avatar
      Merge pull request #4757 from ppt0011/main · 07c2e3d0
      ppt0011 authored
      [doc] explain suitable use case for each plugin
      07c2e3d0
    • Pengtai Xu's avatar
  11. 19 Sep, 2023 2 commits