lorebianchi98
/

Talk2DINOv3-ViTB

Image Segmentation

model_hub_mixin

pytorch_model_hub_mixin

open-vocabulary segmentation

Model card Files Files and versions

lorebianchi98 commited on Oct 16, 2025

Commit

8465f05

·

1 Parent(s): 42c1a72

bug fix

Files changed (1) hide show

dinotext.py +1 -1

dinotext.py CHANGED Viewed

@@ -279,7 +279,7 @@ class DINOText(nn.Module):
         text_embs = text_embs.mean(dim=1).float()
         if type(self.proj) == ProjectionLayer or type(self.proj) == DoubleMLP:
             text_embs = self.proj.project_clip_txt(text_embs)
-        text_embs = us.normalize(text_embs, dim=-1)
         return text_embs

         text_embs = text_embs.mean(dim=1).float()
         if type(self.proj) == ProjectionLayer or type(self.proj) == DoubleMLP:
             text_embs = self.proj.project_clip_txt(text_embs)
+        text_embs = normalize(text_embs, dim=-1)
         return text_embs