jwieting
/

paraphrastic_test

Sentence Similarity

sentence-transformers

feature-extraction

text-embeddings-inference

Model card Files Files and versions

jwieting commited on Nov 14, 2022

Commit

0a985cf

·

1 Parent(s): 7c3a7ce

Update modeling_sim.py

Files changed (1) hide show

modeling_sim.py +2 -2

modeling_sim.py CHANGED Viewed

@@ -1,8 +1,8 @@
 import torch
 from torch import nn
-from transformers import MobileBertPreTrainedModel, MobileBertModel
-class SimModel(MobileBertPreTrainedModel):
     def __init__(self, config):
         super().__init__(config)
         self.config = config

 import torch
 from torch import nn
+from transformers import BertPreTrainedModel
+class SimModel(BertPreTrainedModel):
     def __init__(self, config):
         super().__init__(config)
         self.config = config