Merge branch 'main' of https://huggingface.co/sgugger/my-bert-model
Browse files- 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("
|
| 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)
|