alexnasa commited on
Commit
b1cba9b
·
verified ·
1 Parent(s): 6c84339

Update models/unet_2d_condition.py

Browse files
Files changed (1) hide show
  1. models/unet_2d_condition.py +1 -1
models/unet_2d_condition.py CHANGED
@@ -28,13 +28,13 @@ from diffusers.models.embeddings import (
28
  ImageHintTimeEmbedding,
29
  ImageProjection,
30
  ImageTimeEmbedding,
31
- PositionNet,
32
  TextImageProjection,
33
  TextImageTimeEmbedding,
34
  TextTimeEmbedding,
35
  TimestepEmbedding,
36
  Timesteps,
37
  )
 
38
  from diffusers.models.modeling_utils import ModelMixin
39
  from .unet_2d_blocks import (
40
  UNetMidBlock2DCrossAttn,
 
28
  ImageHintTimeEmbedding,
29
  ImageProjection,
30
  ImageTimeEmbedding,
 
31
  TextImageProjection,
32
  TextImageTimeEmbedding,
33
  TextTimeEmbedding,
34
  TimestepEmbedding,
35
  Timesteps,
36
  )
37
+ from diffusers.models.embeddings import GLIGENTextBoundingboxProjection as PositionNet
38
  from diffusers.models.modeling_utils import ModelMixin
39
  from .unet_2d_blocks import (
40
  UNetMidBlock2DCrossAttn,