Files
2026-07-13 13:24:13 +08:00

5 lines
170 B
Python

from .DiT import DiT_models, DiT
from .Transformer import Transformer_models, Transformer
from .EMA import EMAModel
All_models = {**DiT_models, **Transformer_models}