ONNX
bpawar commited on
Commit
42b4282
1 Parent(s): 2db8f0d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +123 -6
README.md CHANGED
@@ -1,6 +1,123 @@
1
- ---
2
- license: other
3
- license_name: nvidia-open-model-license
4
- license_link: >-
5
- https://developer.download.nvidia.com/licenses/nvidia-open-model-license-agreement-june-2024.pdf
6
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: other
3
+ license_name: nvidia-open-model-license
4
+ license_link: >-
5
+ https://developer.download.nvidia.com/licenses/nvidia-open-model-license-agreement-june-2024.pdf
6
+ base_model:
7
+ - microsoft/Phi-3.5-mini-instruct
8
+ ---
9
+
10
+
11
+ # Phi-3.5-mini-Instruct ONNX INT4
12
+
13
+ ### Model Developer: Microsoft
14
+
15
+ ### Model Description
16
+
17
+ The NVIDIA Phi-3.5-mini-Instruct ONNX INT4 model is the quantized version of the Microsoft Phi-3.5-mini-Instruct model which has 3.8B parameters and is a dense decoder-only Transformer model using the same tokenizer as Phi-3 Mini. It supports 128K context length, therefore the model is capable of several long context tasks including long document/meeting summarization, long document QA, long document information retrieval. For more information, please check [here](https://huggingface.co/microsoft/Phi-3.5-mini-instruct). The NVIDIA Phi-3.5-mini-Instruct ONNX INT4 model is quantized with [TensorRT Model Optimizer](https://github.com/NVIDIA/TensorRT-Model-Optimizer).
18
+
19
+ This model is ready for commercial and research use case.
20
+
21
+ Steps followed to generate this quantized model:
22
+
23
+ * 1. Download Microsoft Phi-3.5-mini-Instruct model in Pytorch bfloat16 format from HuggingFace.
24
+
25
+ * 2. Convert PyTorch model to ONNX FP16 using onnxruntime-genai model builder.
26
+
27
+ * 3. Quantize Phi-3.5-mini-Instruct ONNX FP16 model to Phi-3.5-mini-Instruct ONNX INT4 AWQ model using TensorRT Model Optimizer – Windows.
28
+
29
+ ### Third-Party Community Consideration
30
+ This model is not owned or developed by NVIDIA. This model has been developed and built to a third-party’s requirements for this application and use case; see link to the Non-NVIDIA [Phi3.5-Mini-Instruct Model Card](https://huggingface.co/microsoft/Phi-3.5-mini-instruct).
31
+
32
+
33
+ ### License/Terms of Use:
34
+ GOVERNING TERMS: Use of this model is governed by the NVIDIA Open Model License Agreement (found at https://developer.download.nvidia.com/licenses/nvidia-open-model-license-agreement-june-2024.pdf). ADDITIONAL INFORMATION: Gemma Terms of Use (found at https://ai.google.dev/gemma/terms).
35
+
36
+ ### Reference:
37
+
38
+ [MIT License](https://huggingface.co/microsoft/Phi-3.5-mini-instruct/blob/main/LICENSE )
39
+
40
+ [Phi3.5-mini-Instruct Model Card](https://huggingface.co/microsoft/Phi-3.5-mini-instruct )
41
+
42
+ [Readme](https://huggingface.co/microsoft/Phi-3.5-mini-instruct/blob/main/README.md)
43
+
44
+ ### Model Architecture:
45
+
46
+ Phi-3.5-mini has 3.8B parameters and is a dense decoder-only Transformer model using the same tokenizer as Phi-3 Mini.
47
+
48
+ **Architecture Type:** Transformer <br>
49
+
50
+ **Network Architecture:** Phi3 <br>
51
+
52
+ **Input**
53
+
54
+ * Input Type: Text. It is best suited for prompts using chat format.
55
+
56
+ * Input Format: String
57
+
58
+ * Input Parameters: Sequence (1D)
59
+
60
+ * Other Properties Related to Input: Supports Arabic, Chinese, Czech, Danish, Dutch, English, Finnish, French, German, Hebrew, Hungarian, Italian, Japanese, Korean, Norwegian, Polish, Portuguese, Russian, Spanish, Swedish, Thai, Turkish, Ukrainian
61
+
62
+ **Output**
63
+
64
+ * Output Type: Text
65
+
66
+ * Output Format: String
67
+
68
+ * Output Parameters: Sequence (1D)
69
+
70
+ ## Software Integration:
71
+
72
+ * **Supported Hardware Microarchitecture Compatibility :** Nvidia Ampere and newer GPUs. 6GB or higher VRAM GPUs are recommended. Higher VRAM may be required for larger context length use cases. 
73
+
74
+ * **Supported Operating System(s):**  Windows 
75
+
76
+ ## Model Version(s):  1.0 
77
+
78
+ ## Training, Testing, and Evaluation Datasets:  
79
+ Refer to [Phi3.5-mini-Instruct Model Card](https://huggingface.co/microsoft/Phi-3.5-mini-instruct) for the details.
80
+
81
+ ### Calibration Dataset: cnn_daily mail used for calibration.
82
+
83
+ Link: https://huggingface.co/datasets/abisee/cnn_dailymail
84
+
85
+ * Data Collection Method by dataset: Automated
86
+
87
+ * Labeling Method by dataset: [Unknown]
88
+
89
+ ### Evaluation Dataset:
90
+
91
+ Link: https://people.eecs.berkeley.edu/~hendrycks/data.tar
92
+
93
+ * Data Collection Method by dataset  - Unknown
94
+
95
+ * Labeling Method by dataset  - Not Applicable
96
+
97
+ ## Evaluation Results:
98
+
99
+ **MMLU (5# shots):**
100
+
101
+ With GenAI ORT->DML backend, we got below mentioned accuracy numbers on a desktop RTX 4090 GPU system. 
102
+
103
+ "overall_accuracy": 65.51
104
+
105
+ **Test configuration:**
106
+
107
+ * **GPU:** RTX 4090, RTX 3090.  
108
+
109
+ * **Windows 11:** 23H2
110
+
111
+ * **NVIDIA Graphics driver:** R565 or higher
112
+
113
+ ## Inference:
114
+ Inference Backend: [Onnxruntime-GenAI-DirectML](https://onnxruntime.ai/docs/genai/howto/install.html#directml)
115
+
116
+ We used GenAI ORT->DML backend for inference. The instructions to use this backend are given in readme.txt file available under Files section. 
117
+
118
+
119
+ ## Ethical Considerations:
120
+
121
+ NVIDIA believes Trustworthy AI is a shared responsibility and we have established policies and practices to enable development for a wide array of AI applications.  When downloaded or used in accordance with our terms of service, developers should work with their internal model team to ensure this model meets requirements for the relevant industry and use case and addresses unforeseen product misuse. 
122
+
123
+ Please report security vulnerabilities or NVIDIA AI Concerns [here](https://www.nvidia.com/en-us/support/submit-security-vulnerability/).