EulerHuaji commited on
Commit
5dd746d
·
verified ·
1 Parent(s): d29c252

Update unet/unet_2d_condition.py

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