You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/tests/kit/model_zoo
flybird11111 21aa5de00b
[gemini] hotfix NaN loss while using Gemini + tensor_parallel (#5150)
12 months ago
..
custom [test] merge old components to test to model zoo (#4945) 1 year ago
diffusers [misc] update pre-commit and run all files (#4752) 1 year ago
timm [misc] update pre-commit and run all files (#4752) 1 year ago
torchaudio [misc] update pre-commit and run all files (#4752) 1 year ago
torchrec [test] modify model supporting part of low_level_zero plugin (including correspoding docs) 1 year ago
torchvision [misc] update pre-commit and run all files (#4752) 1 year ago
transformers [gemini] hotfix NaN loss while using Gemini + tensor_parallel (#5150) 12 months ago
__init__.py [test] merge old components to test to model zoo (#4945) 1 year ago
executor.py [test] merge old components to test to model zoo (#4945) 1 year ago
registry.py [shardformer]: support gpt-j, falcon, Mistral and add interleaved pipeline for bert (#5088) 1 year ago