gyopak commited on
Commit
eef36ff
·
verified ·
1 Parent(s): f2855e3

Add repetition penalty to config

Browse files
Files changed (1) hide show
  1. config.json +1 -0
config.json CHANGED
@@ -14,6 +14,7 @@
14
  "rms_norm_eps": 1e-5,
15
  "tie_word_embeddings": true,
16
  "torch_dtype": "bfloat16",
 
17
  "bos_token_id": 2,
18
  "eos_token_id": 3,
19
  "pad_token_id": 1
 
14
  "rms_norm_eps": 1e-5,
15
  "tie_word_embeddings": true,
16
  "torch_dtype": "bfloat16",
17
+ "repetition_penalty": 1.2,
18
  "bos_token_id": 2,
19
  "eos_token_id": 3,
20
  "pad_token_id": 1