Update README.md
Browse files
README.md
CHANGED
@@ -84,8 +84,8 @@ outputs = pipe(
|
|
84 |
prompt,
|
85 |
max_new_tokens=512,
|
86 |
do_sample=True,
|
87 |
-
|
88 |
-
|
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']
|