ColossalAI/tests/kit/model_zoo
flybird11111 21aa5de00b
[gemini] hotfix NaN loss while using Gemini + tensor_parallel (#5150)
* fix

aaa

fix

fix

fix

* fix

* fix

* test ci

* fix ci

fix
2023-12-08 11:10:51 +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 [gemini] hotfix NaN loss while using Gemini + tensor_parallel (#5150) 2023-12-08 11:10:51 +08:00
__init__.py [test] merge old components to test to model zoo (#4945) 2023-10-20 10:35:08 +08:00
executor.py [test] merge old components to test to model zoo (#4945) 2023-10-20 10:35:08 +08:00
registry.py [shardformer]: support gpt-j, falcon, Mistral and add interleaved pipeline for bert (#5088) 2023-11-28 16:54:42 +08:00