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.

Hongxin Liu 1b387ca9fe [shardformer] refactor pipeline grad ckpt config (#5646) ... * [shardformer] refactor pipeline grad ckpt config * [shardformer] refactor pipeline grad ckpt config * [pipeline] fix stage manager		7 months ago
..
chatglm2_6b	[hotfix] fix typo change enabel to enable under colossalai/shardformer/ (#5317)	9 months ago
__init__.py	[shardformer] added development protocol for standardization (#4149)	1 year ago
bert.py	[shardformer] Sequence Parallelism Optimization (#5533)	8 months ago
blip2.py	[shardformer] update colo attention to support custom mask (#5510)	8 months ago
bloom.py	[shardformer] update transformers (#5583)	7 months ago
chatglm2.py	[shardformer] fix chatglm implementation (#5644)	7 months ago
falcon.py	[shardformer] update transformers (#5583)	7 months ago
gpt2.py	[shardformer] update transformers (#5583)	7 months ago
gptj.py	[shardformer] update transformers (#5583)	7 months ago
jit.py	[misc] update pre-commit and run all files (#4752)	1 year ago
llama.py	[shardformer] refactor pipeline grad ckpt config (#5646)	7 months ago
mistral.py	[shardformer] refactor pipeline grad ckpt config (#5646)	7 months ago
opt.py	[shardformer] remove useless code (#5645)	7 months ago
sam.py	[misc] update pre-commit and run all files (#4752)	1 year ago
t5.py	[shardformer] update transformers (#5583)	7 months ago
vit.py	[shardformer] update transformers (#5583)	7 months ago
whisper.py	[shardformer] update transformers (#5583)	7 months ago