justinhl
/

hybrid-qa1

Question Answering

text-generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

justinhl commited on May 14

Commit

4275959

•

1 Parent(s): 5692465

Update config.json

Files changed (1) hide show

config.json +10 -1

config.json CHANGED Viewed

@@ -31,5 +31,14 @@
   "torch_dtype": "float32",
   "transformers_version": "4.40.2",
   "use_cache": true,
-  "vocab_size": 32064
 }

   "torch_dtype": "float32",
   "transformers_version": "4.40.2",
   "use_cache": true,
+  "vocab_size": 32064,
+  "custom_pipelines": {
+    "question-answering": {
+      "impl": "hybrid_pipe.HybridQAPipeline",
+      "pt": [
+        "AutoModelForCausalLM"
+      ],
+      "tf": ["TFAutoModelForCausalLM"]
+    }
+  }
 }