wukong1992 6b30dfb7ce [shardformer] support llama model using shardformer (#3969) ... adjust layer attr		2023-07-04 16:05:01 +08:00
..
__init__.py	[shardformer] init shardformer code structure (#3731)	2023-07-04 16:05:01 +08:00
autopolicy.py	[shardformer] support llama model using shardformer (#3969)	2023-07-04 16:05:01 +08:00
basepolicy.py	[shardformer] Add dropout layer in shard model and refactor policy api (#3949)	2023-07-04 16:05:01 +08:00
bert.py	[shardformer] Add dropout layer in shard model and refactor policy api (#3949)	2023-07-04 16:05:01 +08:00
gpt2.py	[shardformer] Add dropout layer in shard model and refactor policy api (#3949)	2023-07-04 16:05:01 +08:00
llama.py	[shardformer] support llama model using shardformer (#3969)	2023-07-04 16:05:01 +08:00