Hongxin Liu bd18678478 [test] fix gemini checkpoint and gpt test (#4620)		2023-09-05 16:02:23 +08:00
..
__init__.py	…
_utils.py	[shardformer] Pytree fix (#4533)	2023-09-04 17:52:23 +08:00
test_shard_bert.py	[shardformer] Pytree fix (#4533)	2023-09-04 17:52:23 +08:00
test_shard_blip2.py	…
test_shard_bloom.py	…
test_shard_chatglm2.py	…
test_shard_gpt2.py	[test] fix gemini checkpoint and gpt test (#4620)	2023-09-05 16:02:23 +08:00
test_shard_llama.py	…
test_shard_opt.py	…
test_shard_sam.py	…
test_shard_t5.py	…
test_shard_vit.py	…
test_shard_whisper.py	…