Update README.md
Browse files
README.md
CHANGED
@@ -17,8 +17,8 @@ To easily get started with the model, you can use our newly introduced ONNX Runt
|
|
17 |
## ONNX Models
|
18 |
Here are some of the optimized configurations we have added:
|
19 |
|
20 |
-
1. ONNX model for
|
21 |
-
2. ONNX model for
|
22 |
|
23 |
|
24 |
## Performance
|
@@ -34,8 +34,17 @@ ONNX enables you to run your models on-device across CPU, GPU, NPU. With ONNX yo
|
|
34 |
|deepseek-ai_DeepSeek-R1-Distill-Qwen-7B| ONNX | Int4/bfloat16 | CPU | CPU Intel i9 | 3.184 | 20X |
|
35 |
|deepseek-ai_DeepSeek-R1-Distill-Qwen-1.5B| ONNX | Int4 | CPU | CPU Intel i9 | 11.749 | 1.4x |
|
36 |
|
37 |
-
|
38 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
39 |
|
40 |
|
41 |
## Model Description
|
@@ -48,5 +57,6 @@ CPU BUILD SPECS: Onnxruntime-genai==0.6.0-dev, transformers==4.46.2, onnxruntime
|
|
48 |
- **Disclaimer:** Model is only an optimization of the base model, any risk associated with the model is the responsibility of the user of the model. Please verify and test for you scenarios. There may be a slight difference in output from the base model with the optimizations applied. **
|
49 |
|
50 |
## Base Model Information
|
|
|
51 |
See HF links [DeepSeek-R1-Distill-Qwen-1.5B](https://huggingface.co/deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B/) and [DeepSeek-R1-Distill-Qwen-7B](https://huggingface.co/deepseek-ai/DeepSeek-R1-Distill-Qwen-7B/) for details.
|
52 |
|
|
|
17 |
## ONNX Models
|
18 |
Here are some of the optimized configurations we have added:
|
19 |
|
20 |
+
1. ONNX model for CPU and mobile using int4 quantization via RTN.
|
21 |
+
2. ONNX model for GPU using int4 quantization via RTN.
|
22 |
|
23 |
|
24 |
## Performance
|
|
|
34 |
|deepseek-ai_DeepSeek-R1-Distill-Qwen-7B| ONNX | Int4/bfloat16 | CPU | CPU Intel i9 | 3.184 | 20X |
|
35 |
|deepseek-ai_DeepSeek-R1-Distill-Qwen-1.5B| ONNX | Int4 | CPU | CPU Intel i9 | 11.749 | 1.4x |
|
36 |
|
37 |
+
CPU build specs:
|
38 |
+
|
39 |
+
- onnxruntime-genai==0.6.0-dev
|
40 |
+
- transformers==4.46.2
|
41 |
+
- onnxruntime==1.20.01
|
42 |
+
|
43 |
+
CUDA build specs:
|
44 |
+
|
45 |
+
- onnxruntime-genai-cuda==0.6.0-dev
|
46 |
+
- transformers==4.46.2
|
47 |
+
- onnxruntime-gpu==1.20.1
|
48 |
|
49 |
|
50 |
## Model Description
|
|
|
57 |
- **Disclaimer:** Model is only an optimization of the base model, any risk associated with the model is the responsibility of the user of the model. Please verify and test for you scenarios. There may be a slight difference in output from the base model with the optimizations applied. **
|
58 |
|
59 |
## Base Model Information
|
60 |
+
|
61 |
See HF links [DeepSeek-R1-Distill-Qwen-1.5B](https://huggingface.co/deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B/) and [DeepSeek-R1-Distill-Qwen-7B](https://huggingface.co/deepseek-ai/DeepSeek-R1-Distill-Qwen-7B/) for details.
|
62 |
|