Views
No views yet
| モデル | jsquad(1-shot) | jcommonsenseqa(1-shot) | jnli(1-shot) |
|---|---|---|---|
| Kendamarron/Tokara-0.5B-v0.1 | 26.4295 | 0.2663 | 0.5509 |
| Qwen/Qwen1.5-0.5B | 31.3597 | 0.2556 | 0.5534 |
1from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
2
3model = AutoModelForCausalLM.from_pretrained('Kendamarron/Tokara-0.5B-v0.1')
4tokenizer = AutoTokenizer.from_pretrained('Kendamarron/Tokara-0.5B-v0.1')
5
6pipe = pipeline('text-generation', model=model, tokenizer=tokenizer)
7
8prompt = "大規模言語モデルとは、"
9
10print(pipe(prompt, max_length=128, repetition_penalty=1.1, temperature=0.7, top_p=0.95))
11