Update README.md

This commit is contained in:
UnicomAI
2024-04-26 12:58:23 +00:00
parent 3762ffb983
commit 58df9f9ac6

View File

@@ -73,7 +73,7 @@ outputs = pipeline(
prompt, prompt,
max_new_tokens=2048, max_new_tokens=2048,
eos_token_id=terminators, eos_token_id=terminators,
do_sample=False, do_sample=True,
temperature=0.6, temperature=0.6,
top_p=1, top_p=1,
repetition_penalty=1.1 repetition_penalty=1.1