aidadev48 commited on
Commit
4d7323b
1 Parent(s): 847a94a

Update generation_config.json

Browse files

added top K and other gen settings

Files changed (1) hide show
  1. generation_config.json +5 -1
generation_config.json CHANGED
@@ -6,9 +6,13 @@
6
  128008,
7
  128009
8
  ],
9
- "max_length": 131072,
10
  "pad_token_id": 128004,
11
  "temperature": 0.6,
 
 
 
 
12
  "top_p": 0.9,
13
  "transformers_version": "4.46.1"
14
  }
 
6
  128008,
7
  128009
8
  ],
9
+ "max_length": 50,
10
  "pad_token_id": 128004,
11
  "temperature": 0.6,
12
+ "top_k":50,
13
+ "repetition_penalty":1.3,
14
+ "no_repeat_ngram_size":2,
15
+ "eos_token_id":50256,
16
  "top_p": 0.9,
17
  "transformers_version": "4.46.1"
18
  }