UnicomAI commited on
Commit
076a73e
1 Parent(s): 7cf2488

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -73,10 +73,10 @@ outputs = pipeline(
73
  prompt,
74
  max_new_tokens=2048,
75
  eos_token_id=terminators,
76
- do_sample=False,
77
  temperature=0.6,
78
  top_p=1,
79
- repetition_penalty=1.1
80
  )
81
  print(outputs[0]["generated_text"][len(prompt):])
82
  ```
 
73
  prompt,
74
  max_new_tokens=2048,
75
  eos_token_id=terminators,
76
+ do_sample=True,
77
  temperature=0.6,
78
  top_p=1,
79
+ repetition_penalty=1.15
80
  )
81
  print(outputs[0]["generated_text"][len(prompt):])
82
  ```