hf-internal-testing
/

tiny-sdxl-custom-components

StableDiffusionXLPipeline

Inference Endpoints

Model card Files Files and versions Community

sayakpaul HF staff commited on 15 days ago

Commit

0abaa8a

•

1 Parent(s): 16d8d11

fix more

Files changed (1) hide show

unet/my_unet_model.py +3 -3

unet/my_unet_model.py CHANGED Viewed

@@ -34,7 +34,7 @@ from diffusers.models.embeddings import (
     ImageHintTimeEmbedding,
     ImageProjection,
     ImageTimeEmbedding,
-    PositionNet,
     TextImageProjection,
     TextImageTimeEmbedding,
     TextTimeEmbedding,
@@ -42,7 +42,7 @@ from diffusers.models.embeddings import (
     Timesteps,
 )
 from diffusers.models.modeling_utils import ModelMixin
-from diffusers.models.unet_2d_blocks import (
     UNetMidBlock2DCrossAttn,
     UNetMidBlock2DSimpleCrossAttn,
     get_down_block,
@@ -586,7 +586,7 @@ class MyUNetModel(ModelMixin, ConfigMixin, UNet2DConditionLoadersMixin):
                 positive_len = cross_attention_dim[0]
             feature_type = "text-only" if attention_type == "gated" else "text-image"
-            self.position_net = PositionNet(
                 positive_len=positive_len, out_dim=cross_attention_dim, feature_type=feature_type
             )

     ImageHintTimeEmbedding,
     ImageProjection,
     ImageTimeEmbedding,
+    GLIGENTextBoundingboxProjection,
     TextImageProjection,
     TextImageTimeEmbedding,
     TextTimeEmbedding,
     Timesteps,
 )
 from diffusers.models.modeling_utils import ModelMixin
+from diffusers.models.unets.unet_2d_blocks import (
     UNetMidBlock2DCrossAttn,
     UNetMidBlock2DSimpleCrossAttn,
     get_down_block,
                 positive_len = cross_attention_dim[0]
             feature_type = "text-only" if attention_type == "gated" else "text-image"
+            self.position_net = GLIGENTextBoundingboxProjection(
                 positive_len=positive_len, out_dim=cross_attention_dim, feature_type=feature_type
             )