mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-08-12 21:25:53 +00:00
b8e770c832
* [test] add custom models in model zoo * [test] update legacy test * [test] update model zoo * [test] update gemini test * [test] remove components to test
6 lines
213 B
Python
6 lines
213 B
Python
from . import custom, diffusers, timm, torchaudio, torchrec, torchvision, transformers
|
|
from .executor import run_fwd, run_fwd_bwd
|
|
from .registry import model_zoo
|
|
|
|
__all__ = ["model_zoo", "run_fwd", "run_fwd_bwd"]
|