ColossalAI/tests/kit/model_zoo/__init__.py

5 lines
136 B
Python

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