mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-06-22 21:49:08 +00:00
f57d34958b
* pass gpt trace and meta_prop * pass t5 trace and meta_prop * [FX] refactor experimental tracer and adapt it with hf models * pass all mainstream model zoo * fix CI * fix CI * fix CI * fix CI * fix CI * fix CI * fix CI * fix CI * skip tests * fix CI * using packaging version * polish
5 lines
136 B
Python
5 lines
136 B
Python
from . import diffusers, timm, torchaudio, torchrec, torchvision, transformers
|
|
from .registry import model_zoo
|
|
|
|
__all__ = ['model_zoo']
|