8 lines
190 B
Python
8 lines
190 B
Python
from .dit.modeling_dit import DiT_models
|
|
from .latte.modeling_latte import Latte_models
|
|
|
|
Diffusion_models = {}
|
|
Diffusion_models.update(DiT_models)
|
|
Diffusion_models.update(Latte_models)
|
|
|
|
|