Upload unet_2d_condition_woct.py
Browse files
oms_module/unet_2d_condition_woct.py
CHANGED
|
@@ -35,7 +35,7 @@ from diffusers.models.embeddings import (
|
|
| 35 |
Timesteps,
|
| 36 |
)
|
| 37 |
from diffusers.models.modeling_utils import ModelMixin
|
| 38 |
-
from diffusers.models.unet_2d_blocks import (
|
| 39 |
CrossAttnDownBlock2D,
|
| 40 |
CrossAttnUpBlock2D,
|
| 41 |
DownBlock2D,
|
|
|
|
| 35 |
Timesteps,
|
| 36 |
)
|
| 37 |
from diffusers.models.modeling_utils import ModelMixin
|
| 38 |
+
from diffusers.models.unets.unet_2d_blocks import (
|
| 39 |
CrossAttnDownBlock2D,
|
| 40 |
CrossAttnUpBlock2D,
|
| 41 |
DownBlock2D,
|