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

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

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

_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_pipeline.py

[pipeline] All bert models (#4233)

2023-08-15 23:25:14 +08:00

test_shard_bert.py

[shardformer] support lazy init (#4202)

2023-08-15 23:25:14 +08:00

test_shard_bloom_pipeline.py

[pipeline] finish bloom models pipeline and tests (#4223)

2023-08-15 23:25:14 +08:00

test_shard_bloom.py

[shardformer] support lazy init (#4202)

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_gpt2.py

[pipeline] Add Pipeline Forward for GPT2Model Shardformer (#4224)

2023-08-15 23:25:14 +08:00

test_shard_llama_pipeline.py

[pipeline] Llama causal lm and llama for sequence classification pipeline (#4208)

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_opt.py

[shardformer] support lazy init (#4202)

2023-08-15 23:25:14 +08:00

test_shard_t5.py

[shardformer] support lazy init (#4202)

2023-08-15 23:25:14 +08:00

test_shard_vit.py

[shardformer] added embedding gradient check (#4124)

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