kunato commited on
Commit
d43e648
1 Parent(s): 9cc3edf

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -59,10 +59,11 @@ input_ids = tokenizer.apply_chat_template(
59
 
60
  outputs = model.generate(
61
  input_ids,
62
- max_new_tokens=512,
63
  do_sample=True,
64
- temperature=0.4,
65
  top_p=0.9,
 
66
  )
67
  response = outputs[0][input_ids.shape[-1]:]
68
  print(tokenizer.decode(response, skip_special_tokens=True))
 
59
 
60
  outputs = model.generate(
61
  input_ids,
62
+ max_new_tokens=1024,
63
  do_sample=True,
64
+ temperature=0.6,
65
  top_p=0.9,
66
+ repetition_penalty=1.15
67
  )
68
  response = outputs[0][input_ids.shape[-1]:]
69
  print(tokenizer.decode(response, skip_special_tokens=True))