Update modeling_bert.py
Browse files- modeling_bert.py +1 -0
modeling_bert.py
CHANGED
|
@@ -56,6 +56,7 @@ from ...utils import (
|
|
| 56 |
)
|
| 57 |
from .configuration_bert import BertConfig
|
| 58 |
|
|
|
|
| 59 |
|
| 60 |
def softmax_n_shifted_zeros(input: torch.Tensor, n: int, dim=-1) -> torch.Tensor:
|
| 61 |
"""
|
|
|
|
| 56 |
)
|
| 57 |
from .configuration_bert import BertConfig
|
| 58 |
|
| 59 |
+
BertModel.register_for_auto_class("AutoModel")
|
| 60 |
|
| 61 |
def softmax_n_shifted_zeros(input: torch.Tensor, n: int, dim=-1) -> torch.Tensor:
|
| 62 |
"""
|