rishigami commited on
Commit
81340b8
1 Parent(s): 525ad91

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -0
README.md CHANGED
@@ -33,6 +33,8 @@ with torch.no_grad():
33
  max_new_tokens=64,
34
  do_sample=True,
35
  temperature=0.7,
 
 
36
  pad_token_id=tokenizer.pad_token_id,
37
  )
38
 
 
33
  max_new_tokens=64,
34
  do_sample=True,
35
  temperature=0.7,
36
+ top_p=0.9,
37
+ repetition_penalty=1.05,
38
  pad_token_id=tokenizer.pad_token_id,
39
  )
40