| 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) | |