ColossalAI/tests/kit/model_zoo/torchrec
Hongxin Liu 079bf3cb26
[misc] update pre-commit and run all files (#4752)
* [misc] update pre-commit

* [misc] run pre-commit

* [misc] remove useless configuration files

* [misc] ignore cuda for clang-format
2023-09-19 14:20:26 +08:00
..
__init__.py [pipeline] support shardformer for GPT2ForQuestionAnswering & complete pipeline support for GPT2 (#4245) 2023-08-15 23:25:14 +08:00
torchrec.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00