l3cube-pune ravindranayak commited on
Commit
8735155
1 Parent(s): 5afc707

Updated config for different outputs (#1)

Browse files

- Updated config for different outputs (385891908b4fcaf3bf3646435f55615b288687af)


Co-authored-by: Ravindra Nayak <[email protected]>

Files changed (1) hide show
  1. config.json +6 -0
config.json CHANGED
@@ -28,5 +28,11 @@
28
  "torch_dtype": "float32",
29
  "transformers_version": "4.17.0",
30
  "use_cache": true,
 
 
 
 
 
 
31
  "vocab_size": 50257
32
  }
 
28
  "torch_dtype": "float32",
29
  "transformers_version": "4.17.0",
30
  "use_cache": true,
31
+ "task_specific_params": {
32
+ "text-generation": {
33
+ "do_sample": true,
34
+ "max_length": 50
35
+ }
36
+ },
37
  "vocab_size": 50257
38
  }