Fill-Mask
Transformers
PyTorch
Norwegian
Norwegian Bokmål
Norwegian Nynorsk
BERT
NorBERT
Norwegian
encoder
custom_code
Instructions to use ltg/norbert3-base with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use ltg/norbert3-base with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="ltg/norbert3-base", trust_remote_code=True)# Load model directly from transformers import AutoModelForMaskedLM model = AutoModelForMaskedLM.from_pretrained("ltg/norbert3-base", trust_remote_code=True, dtype="auto") - Notebooks
- Google Colab
- Kaggle
Fix compatibility with transformers v5
Browse files- modeling_norbert.py +1 -1
modeling_norbert.py
CHANGED
|
@@ -235,7 +235,7 @@ class NorbertPreTrainedModel(PreTrainedModel):
|
|
| 235 |
nn.init.trunc_normal_(module.weight.data, mean=0.0, std=std, a=-2*std, b=2*std)
|
| 236 |
elif isinstance(module, nn.LayerNorm) and module.weight is not None:
|
| 237 |
module.weight.data.fill_(1.0)
|
| 238 |
-
if module.bias is not None:
|
| 239 |
module.bias.data.zero_()
|
| 240 |
|
| 241 |
|
|
|
|
| 235 |
nn.init.trunc_normal_(module.weight.data, mean=0.0, std=std, a=-2*std, b=2*std)
|
| 236 |
elif isinstance(module, nn.LayerNorm) and module.weight is not None:
|
| 237 |
module.weight.data.fill_(1.0)
|
| 238 |
+
if hasattr(module, "bias") and module.bias is not None:
|
| 239 |
module.bias.data.zero_()
|
| 240 |
|
| 241 |
|