Baizhou Zhang 2a2eacfaf1 [pipeline] support shardformer for GPT2ForQuestionAnswering & complete pipeline support for GPT2 (#4245) ... * change for transformers loggers * add forward for GPT2ForQuestionAnswering * fix assert * fix torchrec test		2023-08-15 23:25:14 +08:00
..
__init__.py	…
_utils.py	[pipeline] All bert models (#4233)	2023-08-15 23:25:14 +08:00
test_pure_pipeline.py	[bugs] hot fix some testing bugs for new models (#4268)	2023-08-15 23:25:14 +08:00
test_shard_bert.py	…
test_shard_bert_pipeline.py	[pipeline] All bert models (#4233)	2023-08-15 23:25:14 +08:00
test_shard_bloom.py	…
test_shard_bloom_pipeline.py	[pipeline] finish bloom models pipeline and tests (#4223)	2023-08-15 23:25:14 +08:00
test_shard_gpt2.py	[pipeline] Add Pipeline Forward for GPT2Model Shardformer (#4224)	2023-08-15 23:25:14 +08:00
test_shard_gpt2_pipeline.py	[pipeline] support shardformer for GPT2ForQuestionAnswering & complete pipeline support for GPT2 (#4245)	2023-08-15 23:25:14 +08:00
test_shard_llama.py	[shardformer] support lazy init (#4202)	2023-08-15 23:25:14 +08:00
test_shard_llama_pipeline.py	…
test_shard_opt.py	…
test_shard_t5.py	…
test_shard_vit.py	…