mirror of https://github.com/hpcaitech/ColossalAI
6ae8ed0407
* [lazyinit] fix shared module * [tests] add lazy init test utils * [tests] add torchvision for lazy init * [lazyinit] fix pre op fn * [lazyinit] handle legacy constructor * [tests] refactor lazy init test models * [tests] refactor lazy init test utils * [lazyinit] fix ops don't support meta * [tests] lazy init test timm models * [lazyinit] fix set data * [lazyinit] handle apex layers * [tests] lazy init test transformers models * [tests] lazy init test torchaudio models * [lazyinit] fix import path * [tests] lazy init test torchrec models * [tests] update torch version in CI * [tests] revert torch version in CI * [tests] skip lazy init test |
||
---|---|---|
.. | ||
diffusers | ||
timm | ||
torchaudio | ||
torchrec | ||
torchvision | ||
transformers | ||
__init__.py | ||
registry.py |