Upload config
Browse files- config.json +4 -25
config.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"_name_or_path": "textminr/tl-flan-t5-base",
|
3 |
"architectures": [
|
4 |
"T5ForConditionalGeneration"
|
5 |
],
|
@@ -26,32 +26,11 @@
|
|
26 |
"relative_attention_max_distance": 128,
|
27 |
"relative_attention_num_buckets": 32,
|
28 |
"task_specific_params": {
|
29 |
-
"
|
30 |
"early_stopping": true,
|
31 |
-
"
|
32 |
-
"max_length": 200,
|
33 |
-
"min_length": 30,
|
34 |
-
"no_repeat_ngram_size": 3,
|
35 |
"num_beams": 4,
|
36 |
-
"prefix": "
|
37 |
-
},
|
38 |
-
"translation_en_to_de": {
|
39 |
-
"early_stopping": true,
|
40 |
-
"max_length": 300,
|
41 |
-
"num_beams": 4,
|
42 |
-
"prefix": "translate English to German: "
|
43 |
-
},
|
44 |
-
"translation_en_to_fr": {
|
45 |
-
"early_stopping": true,
|
46 |
-
"max_length": 300,
|
47 |
-
"num_beams": 4,
|
48 |
-
"prefix": "translate English to French: "
|
49 |
-
},
|
50 |
-
"translation_en_to_ro": {
|
51 |
-
"early_stopping": true,
|
52 |
-
"max_length": 300,
|
53 |
-
"num_beams": 4,
|
54 |
-
"prefix": "translate English to Romanian: "
|
55 |
}
|
56 |
},
|
57 |
"tie_word_embeddings": false,
|
|
|
1 |
{
|
2 |
+
"_name_or_path": "textminr/tl-flan-t5-base-onnx",
|
3 |
"architectures": [
|
4 |
"T5ForConditionalGeneration"
|
5 |
],
|
|
|
26 |
"relative_attention_max_distance": 128,
|
27 |
"relative_attention_num_buckets": 32,
|
28 |
"task_specific_params": {
|
29 |
+
"topic-labeling": {
|
30 |
"early_stopping": true,
|
31 |
+
"max_length": 50,
|
|
|
|
|
|
|
32 |
"num_beams": 4,
|
33 |
+
"prefix": "Provide a topic label: "
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
34 |
}
|
35 |
},
|
36 |
"tie_word_embeddings": false,
|