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

Browse Source

* [shardformer] adapted T5 and LLaMa test to use kit

* polish code

...

This commit is contained in:

Frank Lee

2023-06-21 09:32:46 +08:00

parent 4021b9a8a2

commit 58df720570

24 changed files with 239 additions and 168 deletions

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Download Patch File Download Diff File Expand all files Collapse all files

0

tests/test_shardformer/test_model/__init__.py Normal file