nsorros
/

my_model

Feature Extraction

text-embeddings-inference

Model card Files Files and versions Community

nsorros commited on Mar 29, 2022

Commit

1ca0f10

•

1 Parent(s): ccbf044

add model

Files changed (1) hide show

custom_model.py +1 -1

custom_model.py CHANGED Viewed

@@ -9,5 +9,5 @@ class Model(PreTrainedModel):
         self.model = AutoModel.from_pretrained("bert-base-uncased")
     def forward(self, **inputs):
-        outs = self.model(inputs)
         return outs

         self.model = AutoModel.from_pretrained("bert-base-uncased")
     def forward(self, **inputs):
+        outs = self.model(**inputs)
         return outs