Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
2
3model = AutoModelForCausalLM.from_pretrained('Kendamarron/Yonaguni-460M-v0.1')
4tokenizer = AutoTokenizer.from_pretrained('Kendamarron/Yonaguni-460M-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))