lukeingawesome commited on
Commit
b5f02cc
·
verified ·
1 Parent(s): c589499

Update config to use custom model class

Browse files
Files changed (1) hide show
  1. config.json +2 -2
config.json CHANGED
@@ -1,12 +1,12 @@
1
  {
2
  "_name_or_path": "lukeingawesome/llm2vec4cxr",
3
  "architectures": [
4
- "LlamaBiModel"
5
  ],
6
  "attention_bias": false,
7
  "attention_dropout": 0.0,
8
  "auto_map": {
9
- "AutoModel": "microsoft/LLM2CLIP-Llama-3.2-1B-Instruct-CC-Finetuned--modeling_bidirectional_llama_encoder.LlamaBiModel"
10
  },
11
  "base_model": "microsoft/LLM2CLIP-Llama-3.2-1B-Instruct-CC-Finetuned",
12
  "bos_token_id": 128000,
 
1
  {
2
  "_name_or_path": "lukeingawesome/llm2vec4cxr",
3
  "architectures": [
4
+ "LLM2Vec4CXRModel"
5
  ],
6
  "attention_bias": false,
7
  "attention_dropout": 0.0,
8
  "auto_map": {
9
+ "AutoModel": "modeling_llm2vec4cxr.LLM2Vec4CXRModel"
10
  },
11
  "base_model": "microsoft/LLM2CLIP-Llama-3.2-1B-Instruct-CC-Finetuned",
12
  "bos_token_id": 128000,