Update unet/my_unet_model.py (#4)
Browse files- Update unet/my_unet_model.py (c1f4a0eac1247843fb3ea601a2dd43f588fa467f)
- unet/my_unet_model.py +1 -1
unet/my_unet_model.py
CHANGED
|
@@ -42,7 +42,7 @@ from diffusers.models.embeddings import (
|
|
| 42 |
Timesteps,
|
| 43 |
)
|
| 44 |
from diffusers.models.modeling_utils import ModelMixin
|
| 45 |
-
from diffusers.models.unet_2d_blocks import (
|
| 46 |
UNetMidBlock2DCrossAttn,
|
| 47 |
UNetMidBlock2DSimpleCrossAttn,
|
| 48 |
get_down_block,
|
|
|
|
| 42 |
Timesteps,
|
| 43 |
)
|
| 44 |
from diffusers.models.modeling_utils import ModelMixin
|
| 45 |
+
from diffusers.models.unets.unet_2d_blocks import (
|
| 46 |
UNetMidBlock2DCrossAttn,
|
| 47 |
UNetMidBlock2DSimpleCrossAttn,
|
| 48 |
get_down_block,
|