sgugger commited on
Commit
809ca36
2 Parent(s): 27a58f8 6c97bac

Merge branch 'main' of https://huggingface.co/sgugger/my-bert-model

Browse files
Files changed (1) hide show
  1. modeling.py +1 -1
modeling.py CHANGED
@@ -2,5 +2,5 @@ from transformers import BertModel
2
 
3
  class MyBertModel(BertModel):
4
  def forward(self, *args, **kwargs):
5
- print("Success")
6
  return super().forward(*args, **kwargs)
2
 
3
  class MyBertModel(BertModel):
4
  def forward(self, *args, **kwargs):
5
+ print("This is a custom model, whose codes leaves in https://huggingface.co/sgugger/my-bert-model/edit/main/modeling.py")
6
  return super().forward(*args, **kwargs)