sushanth04ad commited on
Commit
9c9035b
·
verified ·
1 Parent(s): f995f19

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +7 -3
config.json CHANGED
@@ -27,12 +27,16 @@
27
  "summary_type": "cls_index",
28
  "summary_use_proj": true,
29
  "task_specific_params": {
30
- "conversational": {
31
- "max_length": 1000
 
 
 
32
  }
33
  },
 
34
  "torch_dtype": "float32",
35
  "transformers_version": "4.48.3",
36
  "use_cache": true,
37
  "vocab_size": 50257
38
- }
 
27
  "summary_type": "cls_index",
28
  "summary_use_proj": true,
29
  "task_specific_params": {
30
+ "text-generation": {
31
+ "do_sample": true,
32
+ "max_length": 100,
33
+ "top_k": 50,
34
+ "top_p": 0.95
35
  }
36
  },
37
+ "pipeline_tag": "text-generation",
38
  "torch_dtype": "float32",
39
  "transformers_version": "4.48.3",
40
  "use_cache": true,
41
  "vocab_size": 50257
42
+ }