Eurdem commited on
Commit
e440620
1 Parent(s): 64a0c5a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -33,6 +33,6 @@ pipeline = transformers.pipeline(
33
 
34
  messages = [{"role": "user", "content": "Tell me about AI"}]
35
  prompt = pipeline.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
36
- outputs = pipeline(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
37
  print(outputs[0]["generated_text"])
38
  ```
 
33
 
34
  messages = [{"role": "user", "content": "Tell me about AI"}]
35
  prompt = pipeline.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
36
+ outputs = pipeline(prompt, max_new_tokens=1024, do_sample=True, temperature=0.7, top_k=500, top_p=0.95)
37
  print(outputs[0]["generated_text"])
38
  ```