Files
ColossalAI/tests/kit/model_zoo/__init__.py
ver217 14a115000b [tests] model zoo add torchaudio models (#3138)
* [tests] model zoo add torchaudio models

* [tests] refactor torchaudio wavernn

* [tests] refactor fx torchaudio tests
2023-03-15 11:51:16 +08:00

5 lines
126 B
Python

from . import diffusers, timm, torchaudio, torchvision, transformers
from .registry import model_zoo
__all__ = ['model_zoo']