Token Classification
Safetensors
lfm2_span_tagger
hallucination-detection
span-detection
rag
lettucedetect
custom_code
adaamko commited on
Commit
973fa08
·
verified ·
1 Parent(s): 3b7df72

feat: remote-code wrapper so AutoModelForTokenClassification and HallucinationDetector load directly

Browse files
Files changed (1) hide show
  1. config.json +8 -2
config.json CHANGED
@@ -1,8 +1,14 @@
1
  {
2
- "architectures": ["LfmForSpanTagging"],
 
 
 
 
 
3
  "backbone": "LiquidAI/LFM2.5-Encoder-350M",
 
4
  "num_labels": 2,
5
  "id2label": {"0": "supported", "1": "hallucinated"},
6
  "label2id": {"supported": 0, "hallucinated": 1},
7
- "note": "Custom token-classification wrapper; load with the LettuceDetect training script class, see the model card."
8
  }
 
1
  {
2
+ "architectures": ["Lfm2SpanTagger"],
3
+ "model_type": "lfm2_span_tagger",
4
+ "auto_map": {
5
+ "AutoConfig": "modeling_lfm2_span_tagger.Lfm2SpanTaggerConfig",
6
+ "AutoModelForTokenClassification": "modeling_lfm2_span_tagger.Lfm2SpanTagger"
7
+ },
8
  "backbone": "LiquidAI/LFM2.5-Encoder-350M",
9
+ "classifier_dropout": 0.1,
10
  "num_labels": 2,
11
  "id2label": {"0": "supported", "1": "hallucinated"},
12
  "label2id": {"supported": 0, "hallucinated": 1},
13
+ "torch_dtype": "float32"
14
  }