jwieting
/

paraphrastic_test

Sentence Similarity

sentence-transformers

feature-extraction

text-embeddings-inference

Model card Files Files and versions

jwieting commited on Nov 13, 2022

Commit

1eb8ae6

·

1 Parent(s): c5d01af

Update modeling_sim.py

Files changed (1) hide show

modeling_sim.py +1 -1

modeling_sim.py CHANGED Viewed

@@ -12,5 +12,5 @@ class SimModel(MobileBertPreTrainedModel):
     def forward(self, input_ids, attention_mask, token_type_ids, return_dict):
         print(input_ids, attention_mask, token_type_ids)
-        print(inputs_embeds = self.word_embeddings(input_ids).size())
         return self.word_embeddings(input_ids)

     def forward(self, input_ids, attention_mask, token_type_ids, return_dict):
         print(input_ids, attention_mask, token_type_ids)
+        print(self.word_embeddings(input_ids).size())
         return self.word_embeddings(input_ids)