flybird11111 8b7d535977 fix gptj (#5652)		2024-04-26 11:52:27 +08:00
..
__init__.py	[shardformer] init shardformer code structure (#3731)	2023-07-04 16:05:01 +08:00
auto_policy.py	[shardformer] fix chatglm implementation (#5644)	2024-04-25 14:41:17 +08:00
base_policy.py	[shardformer] refactor pipeline grad ckpt config (#5646)	2024-04-25 15:19:30 +08:00
bert.py	[shardformer] refactor embedding resize (#5603)	2024-04-18 16:10:18 +08:00
blip2.py	[shardformer] refactor embedding resize (#5603)	2024-04-18 16:10:18 +08:00
bloom.py	[shardformer] update transformers (#5583)	2024-04-24 22:51:50 +08:00
chatglm2.py	[shardformer] fix chatglm implementation (#5644)	2024-04-25 14:41:17 +08:00
falcon.py	[shardformer] update transformers (#5583)	2024-04-24 22:51:50 +08:00
gpt2.py	[shardformer] update transformers (#5583)	2024-04-24 22:51:50 +08:00
gptj.py	fix gptj (#5652)	2024-04-26 11:52:27 +08:00
llama.py	[shardformer] update transformers (#5583)	2024-04-24 22:51:50 +08:00
mistral.py	[shardformer] update transformers (#5583)	2024-04-24 22:51:50 +08:00
opt.py	[shardformer] update transformers (#5583)	2024-04-24 22:51:50 +08:00
sam.py	[shardformer] update transformers (#5583)	2024-04-24 22:51:50 +08:00
t5.py	[shardformer] refactor embedding resize (#5603)	2024-04-18 16:10:18 +08:00
vit.py	[shardformer, pipeline] add `gradient_checkpointing_ratio` and heterogenous shard policy for llama (#5508)	2024-04-01 11:34:58 +08:00
whisper.py	[shardformer] update transformers (#5583)	2024-04-24 22:51:50 +08:00