Update apdepth/modules/unet_2d_condition.py
Browse files
apdepth/modules/unet_2d_condition.py
CHANGED
|
@@ -45,7 +45,7 @@ from diffusers.models.embeddings import (
|
|
| 45 |
Timesteps,
|
| 46 |
)
|
| 47 |
from diffusers.models.modeling_utils import ModelMixin
|
| 48 |
-
from
|
| 49 |
get_down_block,
|
| 50 |
get_mid_block,
|
| 51 |
get_up_block,
|
|
|
|
| 45 |
Timesteps,
|
| 46 |
)
|
| 47 |
from diffusers.models.modeling_utils import ModelMixin
|
| 48 |
+
from apdepth.modules.unet_2d_blocks import (
|
| 49 |
get_down_block,
|
| 50 |
get_mid_block,
|
| 51 |
get_up_block,
|