Import PositionNet properly from latest diffusers
Browse files
src/models/unet_2d_condition.py
CHANGED
|
@@ -20,13 +20,14 @@ from diffusers.models.embeddings import (
|
|
| 20 |
ImageHintTimeEmbedding,
|
| 21 |
ImageProjection,
|
| 22 |
ImageTimeEmbedding,
|
| 23 |
-
PositionNet,
|
| 24 |
TextImageProjection,
|
| 25 |
TextImageTimeEmbedding,
|
| 26 |
TextTimeEmbedding,
|
| 27 |
TimestepEmbedding,
|
| 28 |
Timesteps,
|
| 29 |
)
|
|
|
|
| 30 |
from diffusers.models.modeling_utils import ModelMixin
|
| 31 |
from diffusers.utils import (
|
| 32 |
USE_PEFT_BACKEND,
|
|
|
|
| 20 |
ImageHintTimeEmbedding,
|
| 21 |
ImageProjection,
|
| 22 |
ImageTimeEmbedding,
|
| 23 |
+
#PositionNet,
|
| 24 |
TextImageProjection,
|
| 25 |
TextImageTimeEmbedding,
|
| 26 |
TextTimeEmbedding,
|
| 27 |
TimestepEmbedding,
|
| 28 |
Timesteps,
|
| 29 |
)
|
| 30 |
+
from diffusers.models.embeddings import GLIGENTextBoundingboxProjection as PositionNet
|
| 31 |
from diffusers.models.modeling_utils import ModelMixin
|
| 32 |
from diffusers.utils import (
|
| 33 |
USE_PEFT_BACKEND,
|