Upload T5ForConditionalGeneration
Browse files- config.json +2 -2
config.json
CHANGED
@@ -1,10 +1,9 @@
|
|
1 |
{
|
2 |
-
"pipeline_tag": "text2text-generation",
|
3 |
-
"base_model": "google/flan-t5-small",
|
4 |
"_name_or_path": "./soulprint-statement-return",
|
5 |
"architectures": [
|
6 |
"T5ForConditionalGeneration"
|
7 |
],
|
|
|
8 |
"classifier_dropout": 0.0,
|
9 |
"d_ff": 1024,
|
10 |
"d_kv": 64,
|
@@ -25,6 +24,7 @@
|
|
25 |
"num_layers": 8,
|
26 |
"output_past": true,
|
27 |
"pad_token_id": 0,
|
|
|
28 |
"relative_attention_max_distance": 128,
|
29 |
"relative_attention_num_buckets": 32,
|
30 |
"task_specific_params": {
|
|
|
1 |
{
|
|
|
|
|
2 |
"_name_or_path": "./soulprint-statement-return",
|
3 |
"architectures": [
|
4 |
"T5ForConditionalGeneration"
|
5 |
],
|
6 |
+
"base_model": "google/flan-t5-small",
|
7 |
"classifier_dropout": 0.0,
|
8 |
"d_ff": 1024,
|
9 |
"d_kv": 64,
|
|
|
24 |
"num_layers": 8,
|
25 |
"output_past": true,
|
26 |
"pad_token_id": 0,
|
27 |
+
"pipeline_tag": "text2text-generation",
|
28 |
"relative_attention_max_distance": 128,
|
29 |
"relative_attention_num_buckets": 32,
|
30 |
"task_specific_params": {
|