fffiloni commited on
Commit
4f546d4
·
verified ·
1 Parent(s): 472797d

Import PositionNet properly from latest diffusers

Browse files
Files changed (1) hide show
  1. src/models/unet_2d_condition.py +2 -1
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,