tkempto1 commited on
Commit
e218d7f
·
verified ·
1 Parent(s): 693f9b6

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +10 -1
config.json CHANGED
@@ -57,5 +57,14 @@
57
  "torch_dtype": "float32",
58
  "transformers_version": "4.40.0",
59
  "use_cache": true,
60
- "vocab_size": 32128
 
 
 
 
 
 
 
 
 
61
  }
 
57
  "torch_dtype": "float32",
58
  "transformers_version": "4.40.0",
59
  "use_cache": true,
60
+ "vocab_size": 32128,
61
+ "custom_pipelines": {
62
+ "question-answering": {
63
+ "impl": "hybrid_pipe2.HybridQAPipeline",
64
+ "pt": [
65
+ "AutoModelForQuestionAnswering"
66
+ ],
67
+ "tf": ["TFAutoModelForQuestionAnswering"]
68
+ }
69
+ }
70
  }