Raghavan
/

indictrans2-indic-en-dist-200M

text2text-generation

Model card Files Files and versions Community

Raghavan commited on Dec 26, 2023

Commit

1094abe

•

1 Parent(s): 2af0a6c

Upload 7 files

Files changed (1) hide show

modeling_indictrans.py +1 -1

modeling_indictrans.py CHANGED Viewed

@@ -615,7 +615,7 @@ class IndicTransEncoder(IndicTransPreTrainedModel):
         sentence_embedding = torch.einsum(
             "i...,i ->i...", sentence_embedding, weights
         )
-        return sentence_embedding
     def forward(
         self,

         sentence_embedding = torch.einsum(
             "i...,i ->i...", sentence_embedding, weights
         )
+        return sentence_embedding.unsqueeze(1)
     def forward(
         self,