tcapelle
/

hallu_scorer

Token Classification

text-generation-inference

Model card Files Files and versions Community

tcapelle commited on Feb 12

Commit

1612e9f

·

verified ·

1 Parent(s): 7da2237

Update config.json

Files changed (1) hide show

config.json +10 -1

config.json CHANGED Viewed

@@ -1,9 +1,18 @@
 {
-  "_name_or_path": "weave_models/vectara",
   "architectures": [
     "T5ForTokenClassification"
   ],
   "classifier_dropout": 0.0,
   "d_ff": 2048,
   "d_kv": 64,
   "d_model": 768,

 {
+  "_name_or_path": "weave_models/new_hallu",
   "architectures": [
     "T5ForTokenClassification"
   ],
   "classifier_dropout": 0.0,
+  "custom_pipelines": {
+    "pair-classification": {
+      "impl": "custom_pipeline.PairTextClassificationPipeline",
+      "pt": [
+        "T5ForTokenClassification"
+      ],
+      "tf": []
+    }
+  },
   "d_ff": 2048,
   "d_kv": 64,
   "d_model": 768,