Views
No views yet
<eos>. Den er en ren basemodell – den fortsetter tekst i læreplanstil og kan ikke svare på spørsmål.1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3tokenizer = AutoTokenizer.from_pretrained("erlingmi/mini-laereplan-gpt")
4modell = AutoModelForCausalLM.from_pretrained("erlingmi/mini-laereplan-gpt")
5
6ids = tokenizer("Skolen skal", return_tensors="pt")["input_ids"]
7resultat = modell.generate(ids, max_new_tokens=60, do_sample=True, temperature=0.7,
8 eos_token_id=tokenizer.eos_token_id)
9print(tokenizer.decode(resultat[0], skip_special_tokens=True).strip())<ukjent> og <eos>)<eos> mellom