|
--- |
|
license: cc-by-nc-sa-3.0 |
|
language: |
|
- 'no' |
|
--- |
|
|
|
Gnerative Pretrained Tranformer with 369M parameters for Norwegian. |
|
|
|
It belongs to NorGLM, a suite of pretrained Norwegian Generative Language Models. The model is based on GPT2 architecture. NorGLM can be used for non-commercial purposes. |
|
|
|
## Datasets |
|
|
|
All models in NorGLM are trained on 200G datasets, nearly 25B tokens, including Norwegian, Denish, Swedish, Germany and English. |
|
|
|
## Run the Model |
|
```python |
|
import torch |
|
from transformers import AutoTokenizer, AutoModelForCausalLM |
|
|
|
model_id = "NorGLM/NorGPT-369M" |
|
tokenizer = AutoTokenizer.from_pretrained(model_id) |
|
model = AutoModelForCausalLM.from_pretrained( |
|
model_id, |
|
device_map='auto', |
|
torch_dtype=torch.bfloat16 |
|
) |
|
|
|
text = "Tom ønsket å gå på barene med venner" |
|
inputs = tokenizer(text, return_tensors="pt") |
|
outputs = model.generate(**inputs, max_new_tokens=20) |
|
``` |
|
|
|
## Note |
|
More training and evaluation details and papers will come soon! |
