Dogacel commited on
Commit
86796a5
·
verified ·
1 Parent(s): c6e6fa4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -6
README.md CHANGED
@@ -20,9 +20,9 @@ base_model:
20
 
21
  This repository uses the weights from the original DeepSeek-OCR 2 and modifies model to support inference on different devices such as CPU and MPS (Apple Metal GPU). By default runs on CPU.
22
 
23
- - [Link to the original DeepSeek-OCR model](https://huggingface.co/deepseek-ai/DeepSeek-OCR-2).
24
 
25
- - [Link to the original DeepSeek-OCR repository](https://github.com/deepseek-ai/DeepSeek-OCR-2)
26
 
27
  <div align="center">
28
  <img src="https://github.com/deepseek-ai/DeepSeek-V2/blob/main/figures/logo.svg?raw=true" width="60%" alt="DeepSeek AI" />
@@ -38,6 +38,8 @@ This repository uses the weights from the original DeepSeek-OCR 2 and modifies m
38
 
39
  ## Usage
40
 
 
 
41
  ```
42
  mamba create -n deepseek-ocr-2 python=3.12.9
43
  mamba activate deepseek-ocr-2
@@ -89,8 +91,8 @@ res = model.infer(
89
  crop_mode = True,
90
  save_results = True,
91
  test_compress = True,
92
- + device = "mps",
93
- + dtype = torch.float16,
94
  )
95
  ```
96
 
@@ -111,7 +113,7 @@ res = model.infer(
111
  crop_mode = True,
112
  save_results = True,
113
  test_compress = True,
114
- + device = "cuda",
115
- + dtype = torch.bfloat16,
116
  )
117
  ```
 
20
 
21
  This repository uses the weights from the original DeepSeek-OCR 2 and modifies model to support inference on different devices such as CPU and MPS (Apple Metal GPU). By default runs on CPU.
22
 
23
+ - [Link to the original DeepSeek-OCR-2 model](https://huggingface.co/deepseek-ai/DeepSeek-OCR-2)
24
 
25
+ - [Link to the original DeepSeek-OCR-2 repository](https://github.com/deepseek-ai/DeepSeek-OCR-2)
26
 
27
  <div align="center">
28
  <img src="https://github.com/deepseek-ai/DeepSeek-V2/blob/main/figures/logo.svg?raw=true" width="60%" alt="DeepSeek AI" />
 
38
 
39
  ## Usage
40
 
41
+ Sample code available at: https://github.com/Dogacel/Universal-DeepSeek-OCR-2
42
+
43
  ```
44
  mamba create -n deepseek-ocr-2 python=3.12.9
45
  mamba activate deepseek-ocr-2
 
91
  crop_mode = True,
92
  save_results = True,
93
  test_compress = True,
94
+ + device = "mps",
95
+ + dtype = torch.float16,
96
  )
97
  ```
98
 
 
113
  crop_mode = True,
114
  save_results = True,
115
  test_compress = True,
116
+ + device = "cuda",
117
+ + dtype = torch.bfloat16,
118
  )
119
  ```