Update README.md
Browse files
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 |
-
+
|
| 93 |
-
+
|
| 94 |
)
|
| 95 |
```
|
| 96 |
|
|
@@ -111,7 +113,7 @@ res = model.infer(
|
|
| 111 |
crop_mode = True,
|
| 112 |
save_results = True,
|
| 113 |
test_compress = True,
|
| 114 |
-
+
|
| 115 |
-
+
|
| 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 |
```
|