Spaces:
Running
on
Zero
Running
on
Zero
Update model/__init__.py
Browse files- model/__init__.py +5 -5
model/__init__.py
CHANGED
@@ -1,10 +1,10 @@
|
|
1 |
-
from
|
2 |
|
3 |
-
from
|
4 |
-
from
|
5 |
-
from
|
6 |
|
7 |
-
from
|
8 |
|
9 |
|
10 |
__all__ = ["CFM", "UNetT", "DiT", "MMDiT", "Trainer"]
|
|
|
1 |
+
from model.cfm import CFM
|
2 |
|
3 |
+
from model.backbones.unett import UNetT
|
4 |
+
from model.backbones.dit import DiT
|
5 |
+
from model.backbones.mmdit import MMDiT
|
6 |
|
7 |
+
from model.trainer import Trainer
|
8 |
|
9 |
|
10 |
__all__ = ["CFM", "UNetT", "DiT", "MMDiT", "Trainer"]
|