kvaishnavi commited on
Commit
b8ffafb
·
verified ·
1 Parent(s): 68d9947

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +14 -4
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 int4 CPU and Mobile: ONNX model for CPU and mobile using int4 quantization via RTN.
21
- 2. ONNX model for int4 CUDA GPU using int4 quantization via RTN.
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
- CUDA BUILD SPECS: Onnxruntime-genai-cuda==0.6.0-dev, transformers==4.46.2, onnxruntime-gpu==1.20.1
38
- CPU BUILD SPECS: Onnxruntime-genai==0.6.0-dev, transformers==4.46.2, onnxruntime==1.20.01
 
 
 
 
 
 
 
 
 
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