selimc commited on
Commit
2571fce
1 Parent(s): 65d33c5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -84,8 +84,8 @@ outputs = pipe(
84
  prompt,
85
  max_new_tokens=512,
86
  do_sample=True,
87
- temperature=0.3,
88
- top_p=0.9
89
  )
90
 
91
  generated_text = outputs[0]['generated_text']
 
84
  prompt,
85
  max_new_tokens=512,
86
  do_sample=True,
87
+ eos_token_id=[pipe.tokenizer.convert_tokens_to_ids("<end_of_turn>"), pipe.tokenizer.eos_token_id],
88
+ temperature=0.67,
89
  )
90
 
91
  generated_text = outputs[0]['generated_text']