77143fb
6c97bac
1ddc92c |
|
from transformers import BertModel
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)
|