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

..

model_zoo

[pipeline] support shardformer for GPT2ForQuestionAnswering & complete pipeline support for GPT2 (#4245)

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

__init__.py

[test] added timm models to test model zoo (#3129)

2023-03-14 14:29:18 +08:00