Views
No views yet
1from transformers import pipeline, LlamaForCausalLM, LlamaTokenizer
2
3model = LlamaForCausalLM.from_pretrained("NYTK/PULI-LlumiX-32K")
4tokenizer = LlamaTokenizer.from_pretrained("NYTK/PULI-LlumiX-32K")
5prompt = "Elmesélek egy történetet a nyelvtechnológiáról."
6generator = pipeline(task="text-generation", model=model, tokenizer=tokenizer)
7
8print(generator(prompt, max_new_tokens=30)[0]["generated_text"])