NorGPT-3B-continue / README.md
NorGLM's picture
Update README.md
564f4d3 verified
|
raw
history blame
982 Bytes
metadata
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

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!