benbaker76's picture
Add DeepSeek-OCR-2 fp32 ONNX export
e4f6fbc verified
Raw History Blame Contribute Delete
715 Bytes
{
"model": "deepseek-ai/DeepSeek-OCR-2",
"dtype": "fp32",
"num_layers": 12,
"num_heads": 10,
"head_dim": 128,
"hidden_size": 1280,
"vocab_size": 129280,
"bos_token_id": 0,
"eos_token_id": 1,
"image_token_id": 128815,
"base_size": 1024,
"crop_size": 768,
"base_tokens": 256,
"crop_tokens": 144,
"min_crops": 2,
"max_crops": 6,
"prompts": {
"markdown": {
"text": "<image>\n<|grounding|>Convert the document to markdown.",
"prefix_ids": [],
"suffix_ids": [
201,
128820,
21842,
270,
4940,
304,
2121,
7919,
16
]
},
"ocr": {
"text": "<image>\nFree OCR.",
"prefix_ids": [],
"suffix_ids": [
201,
21431,
126041,
16
]
}
}
}