Saurabh Kulshreshtha commited on
Commit
71a28d4
1 Parent(s): d23e922

added top-p and top-k sampling

Browse files
Files changed (1) hide show
  1. config.json +10 -1
config.json CHANGED
@@ -20,8 +20,17 @@
20
  "output_past": true,
21
  "pad_token_id": 0,
22
  "relative_attention_num_buckets": 32,
 
 
 
 
 
 
 
 
 
23
  "tie_word_embeddings": false,
24
  "transformers_version": "4.8.1",
25
  "use_cache": true,
26
- "vocab_size": 32128
27
  }
20
  "output_past": true,
21
  "pad_token_id": 0,
22
  "relative_attention_num_buckets": 32,
23
+ "task_specific_params": {
24
+ "text2text-generation": {
25
+ "do_sample": true,
26
+ "max_length": 50,
27
+ "top_k": 50,
28
+ "top_p": 0.95,
29
+ "num_return_sequences": 10
30
+ }
31
+ },
32
  "tie_word_embeddings": false,
33
  "transformers_version": "4.8.1",
34
  "use_cache": true,
35
+ "vocab_size": 32128
36
  }