Update config.json
Browse files- config.json +10 -1
config.json
CHANGED
@@ -31,5 +31,14 @@
|
|
31 |
"torch_dtype": "float32",
|
32 |
"transformers_version": "4.40.2",
|
33 |
"use_cache": true,
|
34 |
-
"vocab_size": 32064
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
35 |
}
|
|
|
31 |
"torch_dtype": "float32",
|
32 |
"transformers_version": "4.40.2",
|
33 |
"use_cache": true,
|
34 |
+
"vocab_size": 32064,
|
35 |
+
"custom_pipelines": {
|
36 |
+
"question-answering": {
|
37 |
+
"impl": "hybrid_pipe.HybridQAPipeline",
|
38 |
+
"pt": [
|
39 |
+
"AutoModelForCausalLM"
|
40 |
+
],
|
41 |
+
"tf": ["TFAutoModelForCausalLM"]
|
42 |
+
}
|
43 |
+
}
|
44 |
}
|