Spaces:
Running
on
Zero
Running
on
Zero
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,
|