5 lines
170 B
Python
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} |