sgugger
/

my-bert-model

Feature Extraction

Model card Files Files and versions Community

sgugger commited on Oct 4, 2021

Commit

6c97bac

•

1 Parent(s): f6b867d

Update with custom url print

Files changed (1) hide show

modeling.py +1 -1

modeling.py CHANGED Viewed

@@ -2,5 +2,5 @@ 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("This is a custom model, whose codes leaves in https://huggingface.co/sgugger/my-bert-model/edit/main/modeling.py")
         return super().forward(*args, **kwargs)