sgugger
/

my-bert-model

Feature Extraction

text-embeddings-inference

Model card Files Files and versions Community

sgugger commited on Sep 2, 2021

Commit

77143fb

·

1 Parent(s): 793ec33

add model

Files changed (1) hide show

modeling.py +6 -0

modeling.py ADDED Viewed

	@@ -0,0 +1,6 @@

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