BabaK07 commited on
Commit
c00ee2c
·
verified ·
1 Parent(s): b127e5d

FIX: Add proper config.json with from_pretrained support

Browse files
Files changed (1) hide show
  1. config.json +11 -8
config.json CHANGED
@@ -1,14 +1,17 @@
1
  {
2
  "architectures": [
3
- "WorkingQwenOCRModel"
4
  ],
5
- "model_type": "custom-qwen-ocr",
6
  "base_model": "Qwen/Qwen2-VL-2B-Instruct",
7
- "custom_ocr_heads": true,
8
- "qwen_hidden_size": 1536,
9
- "torch_dtype": "float16",
10
- "transformers_version": "4.37.0",
11
  "auto_map": {
12
- "AutoModel": "modeling_custom_ocr.WorkingQwenOCRModel"
13
- }
 
 
 
14
  }
 
1
  {
2
  "architectures": [
3
+ "FixedTextractAI"
4
  ],
5
+ "model_type": "textract",
6
  "base_model": "Qwen/Qwen2-VL-2B-Instruct",
7
+ "hidden_size": 1536,
8
+ "vocab_size": 152064,
9
+ "torch_dtype": "float32",
10
+ "transformers_version": "4.40.0",
11
  "auto_map": {
12
+ "AutoModel": "modeling_textract.FixedTextractAI",
13
+ "AutoConfig": "modeling_textract.TextractConfig"
14
+ },
15
+ "_name_or_path": "BabaK07/textract-ai",
16
+ "custom_model": true
17
  }