fragata commited on
Commit
a589894
1 Parent(s): 8027148

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -37,5 +37,5 @@ tokenizer = LlamaTokenizer.from_pretrained("NYTK/PULI-LlumiX-32K")
37
  prompt = "Elmesélek egy történetet a nyelvtechnológiáról."
38
  generator = pipeline(task="text-generation", model=model, tokenizer=tokenizer)
39
 
40
- print(generator(prompt)[0]["generated_text"])
41
  ```
 
37
  prompt = "Elmesélek egy történetet a nyelvtechnológiáról."
38
  generator = pipeline(task="text-generation", model=model, tokenizer=tokenizer)
39
 
40
+ print(generator(prompt, max_new_tokens=30)[0]["generated_text"])
41
  ```