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 |
}
|