ai-sage
/

Giga-Embeddings-instruct

Feature Extraction

Model card Files Files and versions Community

ekolodin commited on 6 days ago

Commit

4e5f0d1

•

1 Parent(s): a58b8cb

Update modeling_gigarembed.py

Files changed (1) hide show

modeling_gigarembed.py +1 -1

modeling_gigarembed.py CHANGED Viewed

@@ -414,7 +414,7 @@ class GigarEmbedModel(PreTrainedModel):
         embeds = self.latent_attention_model(
             outputs.last_hidden_state,
-            attention_mask,
         )
         if not return_dict:
             return (embeds,)

         embeds = self.latent_attention_model(
             outputs.last_hidden_state,
+            pool_mask,
         )
         if not return_dict:
             return (embeds,)