Files
ColossalAI/tests/kit/model_zoo/__init__.py
2023-03-15 05:46:04 +00:00

6 lines
137 B
Python

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