sgugger
/

my-bert-model

Feature Extraction

text-embeddings-inference

Model card Files Files and versions Community

sgugger commited on Sep 2, 2021

Commit

1ddc92c

·

1 Parent(s): 0ca266b

Fix syntax error

Files changed (1) hide show

modeling.py +1 -1

modeling.py CHANGED Viewed

@@ -3,4 +3,4 @@ from transformers import BertModel
 class MyBertModel(BertModel):
     def forward(self, *args, **kwargs):
         print("Success")
-        return super.forward(*args, **kwargs)

 class MyBertModel(BertModel):
     def forward(self, *args, **kwargs):
         print("Success")
+        return super().forward(*args, **kwargs)