ayjays132 commited on
Commit
a582ebc
1 Parent(s): e47e221

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +13 -6
config.json CHANGED
@@ -22,16 +22,23 @@
22
  "summary_type": "cls_index",
23
  "summary_use_proj": true,
24
  "task_specific_params": {
25
- "text-generation": {
26
- "do_sample": true,
27
- "max_length": 1024
 
 
 
 
 
 
 
28
  }
29
  },
30
  "transformers_version": "4.34.0",
31
  "language": ["en"],
32
- "tags": ["text-generation-inference"],
33
- "metrics": ["accuracy"],
34
- "pipeline_tag": "text-generation",
35
  "library_name": "transformers",
36
  "datasets": ["vicgalle/alpaca-gpt4"],
37
  "license": "apache-2.0",
 
22
  "summary_type": "cls_index",
23
  "summary_use_proj": true,
24
  "task_specific_params": {
25
+ "conversational": {
26
+ "max_length": 1024,
27
+ "min_length": 20,
28
+ "length_penalty": 1.5,
29
+ "num_beams": 5,
30
+ "early_stopping": true,
31
+ "no_repeat_ngram_size": 3,
32
+ "temperature": 0.7,
33
+ "top_k": 50,
34
+ "top_p": 0.9
35
  }
36
  },
37
  "transformers_version": "4.34.0",
38
  "language": ["en"],
39
+ "tags": ["conversational"],
40
+ "metrics": ["perplexity", "accuracy"],
41
+ "pipeline_tag": "conversational",
42
  "library_name": "transformers",
43
  "datasets": ["vicgalle/alpaca-gpt4"],
44
  "license": "apache-2.0",