MANSTAGE commited on
Commit
6360dbc
·
verified ·
1 Parent(s): 457751e

Fix config.json with correct task type

Browse files
Files changed (1) hide show
  1. config.json +12 -1
config.json CHANGED
@@ -31,7 +31,18 @@
31
  "do_sample": true,
32
  "max_length": 2048,
33
  "temperature": 0.7,
34
- "top_p": 0.9
 
35
  }
 
 
 
 
 
 
 
 
 
 
36
  }
37
  }
 
31
  "do_sample": true,
32
  "max_length": 2048,
33
  "temperature": 0.7,
34
+ "top_p": 0.9,
35
+ "repetition_penalty": 1.1
36
  }
37
+ },
38
+ "generation_config": {
39
+ "do_sample": true,
40
+ "max_length": 2048,
41
+ "temperature": 0.7,
42
+ "top_p": 0.9,
43
+ "repetition_penalty": 1.1,
44
+ "pad_token_id": 0,
45
+ "eos_token_id": 2,
46
+ "bos_token_id": 1
47
  }
48
  }