NorGPT-3B-continue / README.md
NorGLM's picture
Update README.md
564f4d3 verified
|
raw
history blame contribute delete
No virus
982 Bytes
---
license: cc-by-nc-sa-4.0
language:
- 'no'
---
Gnerative Pretrained Tranformer with 3 Billion parameters for Norwegian. The model is continue trained using NorGPT-3B model on a selective documents from the pretraining dataset, which includes news articles, parlamentary speech, books and govermental reports.
It belongs to NorGLM, a suite of pretrained Norwegian Generative Language Models. NorGLM can be used for non-commercial purposes.
## Run the Model
```python
import torch
from transformers import AutoTokenizer, AutoModelForCausalLM
model_id = "NorGLM/NorGPT-3B-continue"
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!