Update unet/unet_2d_condition.py
Browse files
unet/unet_2d_condition.py
CHANGED
|
@@ -35,15 +35,21 @@ from diffusers.models.embeddings import (
|
|
| 35 |
ImageHintTimeEmbedding,
|
| 36 |
ImageProjection,
|
| 37 |
ImageTimeEmbedding,
|
| 38 |
-
PositionNet,
|
| 39 |
TextImageProjection,
|
| 40 |
TextImageTimeEmbedding,
|
| 41 |
TextTimeEmbedding,
|
| 42 |
TimestepEmbedding,
|
| 43 |
Timesteps,
|
| 44 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 45 |
from diffusers.models.modeling_utils import ModelMixin
|
| 46 |
-
from diffusers.models.unet_2d_blocks import (
|
| 47 |
UNetMidBlock2D,
|
| 48 |
UNetMidBlock2DCrossAttn,
|
| 49 |
UNetMidBlock2DSimpleCrossAttn,
|
|
|
|
| 35 |
ImageHintTimeEmbedding,
|
| 36 |
ImageProjection,
|
| 37 |
ImageTimeEmbedding,
|
| 38 |
+
# PositionNet,
|
| 39 |
TextImageProjection,
|
| 40 |
TextImageTimeEmbedding,
|
| 41 |
TextTimeEmbedding,
|
| 42 |
TimestepEmbedding,
|
| 43 |
Timesteps,
|
| 44 |
)
|
| 45 |
+
import diffusers
|
| 46 |
+
if diffusers.__version__ >'0.25':
|
| 47 |
+
from diffusers.models.embeddings import GLIGENTextBoundingboxProjection as PositionNet
|
| 48 |
+
else:
|
| 49 |
+
from diffusers.models.embeddings import PositionNet
|
| 50 |
+
|
| 51 |
from diffusers.models.modeling_utils import ModelMixin
|
| 52 |
+
from diffusers.models.unets.unet_2d_blocks import (
|
| 53 |
UNetMidBlock2D,
|
| 54 |
UNetMidBlock2DCrossAttn,
|
| 55 |
UNetMidBlock2DSimpleCrossAttn,
|