SykoSLM commited on
Commit
8074022
·
verified ·
1 Parent(s): e9608ee

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -102,10 +102,10 @@ with torch.no_grad():
102
  outputs = model.generate(
103
  **inputs,
104
  max_new_tokens=200,
105
- temperature=0.7,
106
  top_p=0.9,
107
  do_sample=True,
108
- repetition_penalty=1.1,
109
  )
110
 
111
  print(tokenizer.decode(outputs[0], skip_special_tokens=False))
 
102
  outputs = model.generate(
103
  **inputs,
104
  max_new_tokens=200,
105
+ temperature=0.3,
106
  top_p=0.9,
107
  do_sample=True,
108
+ repetition_penalty=1.16,
109
  )
110
 
111
  print(tokenizer.decode(outputs[0], skip_special_tokens=False))