You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/shardformer/modeling
Jianghai ef4c14a5e2
[Inference] Fix bug in ChatGLM2 Tensor Parallelism (#5014)
1 year ago
..
chatglm2_6b [Inference] Fix bug in ChatGLM2 Tensor Parallelism (#5014) 1 year ago
__init__.py [shardformer] added development protocol for standardization (#4149) 1 year ago
bert.py [misc] update pre-commit and run all files (#4752) 1 year ago
blip2.py [misc] update pre-commit and run all files (#4752) 1 year ago
bloom.py [misc] update pre-commit and run all files (#4752) 1 year ago
chatglm2.py [misc] update pre-commit and run all files (#4752) 1 year ago
gpt2.py [misc] update pre-commit and run all files (#4752) 1 year ago
jit.py [misc] update pre-commit and run all files (#4752) 1 year ago
llama.py [misc] update pre-commit and run all files (#4752) 1 year ago
opt.py [misc] update pre-commit and run all files (#4752) 1 year ago
sam.py [misc] update pre-commit and run all files (#4752) 1 year ago
t5.py [misc] update pre-commit and run all files (#4752) 1 year ago
vit.py [hotfix] fix torch 2.0 compatibility (#4936) 1 year ago
whisper.py [misc] update pre-commit and run all files (#4752) 1 year ago