Spaces:

LukeOLuck
/

MiniLM_HC3_Semantic_Ranking

Sleeping

LukeOLuck commited on Dec 18, 2023

Commit

2462e43

1 Parent(s): 083997e

Update forward

Files changed (1) hide show

model.py CHANGED Viewed

@@ -8,7 +8,7 @@ class AutoModelForSentenceEmbedding(nn.Module):
     self.model = model
-  def forward(self, **kwargs)
     model_output = self.model(**kwargs)
     embeddings = self.mean_pooling(model_output, kwargs['attention_mask'])
     embeddings = torch.nn.functional.normalize(embeddings, p=2, dim=1)

     self.model = model
+  def forward(self, **kwargs):
     model_output = self.model(**kwargs)
     embeddings = self.mean_pooling(model_output, kwargs['attention_mask'])
     embeddings = torch.nn.functional.normalize(embeddings, p=2, dim=1)