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)# pip install -U transformers accelerate # Load model directly from transformers import AutoModelForMaskedLM model = AutoModelForMaskedLM.from_pretrained("ltg/norbert3-base", trust_remote_code=True, device_map="auto") - Notebooks
- Google Colab
- Kaggle
Fix compatibility with transformers v5
Browse files- modeling_norbert.py +2 -2
modeling_norbert.py
CHANGED
|
@@ -227,9 +227,9 @@ class NorbertPreTrainedModel(PreTrainedModel):
|
|
| 227 |
if isinstance(module, nn.Linear) or isinstance(module, nn.Embedding):
|
| 228 |
nn.init.trunc_normal_(module.weight, mean=0.0, std=std, a=-2*std, b=2*std)
|
| 229 |
elif isinstance(module, nn.LayerNorm) and module.weight is not None:
|
| 230 |
-
|
| 231 |
if hasattr(module, "bias") and module.bias is not None:
|
| 232 |
-
|
| 233 |
|
| 234 |
|
| 235 |
class NorbertModel(NorbertPreTrainedModel):
|
|
|
|
| 227 |
if isinstance(module, nn.Linear) or isinstance(module, nn.Embedding):
|
| 228 |
nn.init.trunc_normal_(module.weight, mean=0.0, std=std, a=-2*std, b=2*std)
|
| 229 |
elif isinstance(module, nn.LayerNorm) and module.weight is not None:
|
| 230 |
+
nn.init.ones_(module.weight)
|
| 231 |
if hasattr(module, "bias") and module.bias is not None:
|
| 232 |
+
nn.init.zeros_(module.bias)
|
| 233 |
|
| 234 |
|
| 235 |
class NorbertModel(NorbertPreTrainedModel):
|