Frank Lee b1c2901530 [shardformer] supported bloom model (#4098)

2023-07-04 16:05:01 +08:00

..

__init__.py

[shardformer] supported bloom model (#4098)

2023-07-04 16:05:01 +08:00

albert.py

[test] added transformers models to test model zoo (#3135)

2023-03-15 11:26:10 +08:00

bert.py

support kit use for bert/gpt test (#4055)

2023-07-04 16:05:01 +08:00

bloom.py

[shardformer] supported bloom model (#4098)

2023-07-04 16:05:01 +08:00

gpt.py

[shardformer] supported bloom model (#4098)

2023-07-04 16:05:01 +08:00

llama.py

[shardformer] adapted T5 and LLaMa test to use kit (#4049)

2023-07-04 16:05:01 +08:00

opt.py

[shardformer] shardformer support opt models (#4091)

2023-07-04 16:05:01 +08:00

t5.py

[shardformer] adapted T5 and LLaMa test to use kit (#4049)

2023-07-04 16:05:01 +08:00