ColossalAI/tests/kit/model_zoo
Frank Lee d69cd2eb89
[workflow] fixed oom tests (#5275)
* [workflow] fixed oom tests

* polish

* polish

* polish
2024-01-16 18:55:13 +08:00
..
custom [test] merge old components to test to model zoo (#4945) 2023-10-20 10:35:08 +08:00
diffusers [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
timm [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
torchaudio [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
torchrec [test] modify model supporting part of low_level_zero plugin (including correspoding docs) 2023-10-05 15:10:31 +08:00
torchvision [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
transformers [workflow] fixed oom tests (#5275) 2024-01-16 18:55:13 +08:00
__init__.py [workflow] fixed build CI (#5240) 2024-01-10 22:34:16 +08:00
executor.py [test] merge old components to test to model zoo (#4945) 2023-10-20 10:35:08 +08:00
registry.py [workflow] fixed oom tests (#5275) 2024-01-16 18:55:13 +08:00