dfurman commited on
Commit
d359a0b
1 Parent(s): cf59add

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -97,7 +97,7 @@ prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_
97
  # print("***Prompt:\n", prompt)
98
 
99
  outputs = pipeline(
100
- prompt, max_new_tokens=1000, do_sample=True, temperature=0.01, top_k=50, top_p=0.95
101
  )
102
  print("***Generation:")
103
  print(outputs[0]["generated_text"][len(prompt) :])
 
97
  # print("***Prompt:\n", prompt)
98
 
99
  outputs = pipeline(
100
+ prompt, max_new_tokens=1000, do_sample=True, temperature=0.7, top_k=50, top_p=0.95
101
  )
102
  print("***Generation:")
103
  print(outputs[0]["generated_text"][len(prompt) :])