Upload TestEnsembleQAPipeline
Browse files- config.json +2 -2
- generation_config.json +1 -1
config.json
CHANGED
@@ -6,7 +6,7 @@
|
|
6 |
"classifier_dropout": 0.0,
|
7 |
"custom_pipelines": {
|
8 |
"test-qa": {
|
9 |
-
"impl": "
|
10 |
"pt": [
|
11 |
"AutoModelForSeq2SeqLM"
|
12 |
],
|
@@ -67,7 +67,7 @@
|
|
67 |
},
|
68 |
"tie_word_embeddings": false,
|
69 |
"torch_dtype": "float32",
|
70 |
-
"transformers_version": "4.40.
|
71 |
"use_cache": true,
|
72 |
"vocab_size": 32128
|
73 |
}
|
|
|
6 |
"classifier_dropout": 0.0,
|
7 |
"custom_pipelines": {
|
8 |
"test-qa": {
|
9 |
+
"impl": "__main__.TestEnsembleQAPipeline",
|
10 |
"pt": [
|
11 |
"AutoModelForSeq2SeqLM"
|
12 |
],
|
|
|
67 |
},
|
68 |
"tie_word_embeddings": false,
|
69 |
"torch_dtype": "float32",
|
70 |
+
"transformers_version": "4.40.2",
|
71 |
"use_cache": true,
|
72 |
"vocab_size": 32128
|
73 |
}
|
generation_config.json
CHANGED
@@ -3,5 +3,5 @@
|
|
3 |
"decoder_start_token_id": 0,
|
4 |
"eos_token_id": 1,
|
5 |
"pad_token_id": 0,
|
6 |
-
"transformers_version": "4.40.
|
7 |
}
|
|
|
3 |
"decoder_start_token_id": 0,
|
4 |
"eos_token_id": 1,
|
5 |
"pad_token_id": 0,
|
6 |
+
"transformers_version": "4.40.2"
|
7 |
}
|