Update model/___init__.py
Browse files- model/___init__.py +1 -1
model/___init__.py
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
from model.
|
| 2 |
|
| 3 |
from model.backbones.unett import UNetT
|
| 4 |
from model.backbones.dit import DiT
|
|
|
|
| 1 |
+
from model.CFM import CFM
|
| 2 |
|
| 3 |
from model.backbones.unett import UNetT
|
| 4 |
from model.backbones.dit import DiT
|