YiYiXu commited on
Commit
29f2cbe
1 Parent(s): 9389b73

Update unet/my_unet_model.py (#1)

Browse files

- Update unet/my_unet_model.py (8ec3bf760930746b3df00aa78bd33dcb6ebff992)

Files changed (1) hide show
  1. unet/my_unet_model.py +2 -2
unet/my_unet_model.py CHANGED
@@ -34,7 +34,7 @@ from diffusers.models.embeddings import (
34
  ImageHintTimeEmbedding,
35
  ImageProjection,
36
  ImageTimeEmbedding,
37
- PositionNet,
38
  TextImageProjection,
39
  TextImageTimeEmbedding,
40
  TextTimeEmbedding,
@@ -586,7 +586,7 @@ class MyUNetModel(ModelMixin, ConfigMixin, UNet2DConditionLoadersMixin):
586
  positive_len = cross_attention_dim[0]
587
 
588
  feature_type = "text-only" if attention_type == "gated" else "text-image"
589
- self.position_net = PositionNet(
590
  positive_len=positive_len, out_dim=cross_attention_dim, feature_type=feature_type
591
  )
592
 
 
34
  ImageHintTimeEmbedding,
35
  ImageProjection,
36
  ImageTimeEmbedding,
37
+ GLIGENTextBoundingboxProjection,
38
  TextImageProjection,
39
  TextImageTimeEmbedding,
40
  TextTimeEmbedding,
 
586
  positive_len = cross_attention_dim[0]
587
 
588
  feature_type = "text-only" if attention_type == "gated" else "text-image"
589
+ self.position_net = GLIGENTextBoundingboxProjection(
590
  positive_len=positive_len, out_dim=cross_attention_dim, feature_type=feature_type
591
  )
592