emrekuruu commited on
Commit
86614aa
·
verified ·
1 Parent(s): 62087a8

Set model_type=retrievalrouter

Browse files
Files changed (1) hide show
  1. config.json +5 -5
config.json CHANGED
@@ -1,10 +1,10 @@
1
  {
2
  "architectures": [
3
- "IRouterLMModel"
4
  ],
5
  "auto_map": {
6
- "AutoConfig": "configuration_irouterlm.IRouterLMConfig",
7
- "AutoModel": "modeling_irouterlm.IRouterLMModel"
8
  },
9
  "base_model_name": "Qwen/Qwen3-0.6B-Base",
10
  "classifier_dropout": 0.1,
@@ -24,7 +24,7 @@
24
  "LABEL_3": 3,
25
  "LABEL_4": 4
26
  },
27
- "model_type": "irouterlm",
28
  "strategy_names": [
29
  "MULTIMODAL_RERANK",
30
  "MULTIMODAL-SINGLE",
@@ -33,4 +33,4 @@
33
  "BM25"
34
  ],
35
  "transformers_version": "5.13.0"
36
- }
 
1
  {
2
  "architectures": [
3
+ "RetrievalRouterModel"
4
  ],
5
  "auto_map": {
6
+ "AutoConfig": "configuration_retrievalrouter.RetrievalRouterConfig",
7
+ "AutoModel": "modeling_retrievalrouter.RetrievalRouterModel"
8
  },
9
  "base_model_name": "Qwen/Qwen3-0.6B-Base",
10
  "classifier_dropout": 0.1,
 
24
  "LABEL_3": 3,
25
  "LABEL_4": 4
26
  },
27
+ "model_type": "retrievalrouter",
28
  "strategy_names": [
29
  "MULTIMODAL_RERANK",
30
  "MULTIMODAL-SINGLE",
 
33
  "BM25"
34
  ],
35
  "transformers_version": "5.13.0"
36
+ }