license: other
license_name: nvidia-open-model-license
license_link: >-
https://developer.download.nvidia.com/licenses/nvidia-open-model-license-agreement-june-2024.pdf
base_model:
- mistralai/Mistral-7B-Instruct-v0.3
Mistral-7B-Instruct-v0.3 ONNX INT4
Model Developer: Mistralai
Model Description
The Mistral-7B-Instruct-v0.3 ONNX INT4 model is the quantized version of the Mistral-7B-Instruct-v0.3 model, which is an instruct fine-tuned version of the Mistral-7B-v0.3 model used for text generation and question answering. Quantization is done with TensorRT Model Optimizer-Windows.
This model is ready for commercial use.
Steps followed to generate this quantized model:
- Download Mistral-7B-Instruct-v0.3 model in Pytorch bfloat16 format from HuggingFace.
- Convert PyTorch model to ONNX FP16 using onnxruntime-genai model builder.
- Quantize Mistral-7B-Instruct-v0.3 ONNX FP16 model to Mistral-7B-Instruct-v0.3 ONNX INT4 AWQ model using TensorRT Model Optimizer – Windows.
Third-Party Community Consideration
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 Non-NVIDIA Mistral-7B-Instruct-v0.3 Model card.
License/Terms of Use:
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: Apache License, Version 2.0 (found at https://huggingface.co/datasets/choosealicense/licenses/blob/main/markdown/apache-2.0.md ).
Model Architecture:
The Mistral-7B-Instruct-v0.3 Large Language Model (LLM) is an instruct fine-tuned version of the Mistral-7B-v0.3.
Architecture Type: Transformer
Network Architecture: Mistral-7B
Input
Input Type: Text
Input Format: String
Input Parameters: Sequence (1D)
Other Properties Related to Input: Supports English, German, French, Italian, Portuguese, Hindi, Spanish, and Thai
Output
Output Type: Text
Output Format: String
Output Parameters: Sequence (1D)
Software Integration:
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.
Supported Operating System(s): Windows
Model Version(s): 1.0
Training, Testing, and Evaluation Datasets:
Refer to https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.3 for the details.
Calibration Dataset: cnn_dailymail used for calibration.
Link: https://huggingface.co/datasets/abisee/cnn_dailymail
Data Collection Method by dataset: Automated
Labeling Method by dataset: [Unknown]
Evaluation Dataset:
Link: https://people.eecs.berkeley.edu/~hendrycks/data.tar
Data Collection Method by dataset - Unknown
Labeling Method by dataset - Not Applicable
Evaluation Results:
MMLU (5# shots):
With GenAI ORT->DML backend, we got below mentioned accuracy numbers on a desktop RTX 4090 GPU system.
"overall_accuracy": 60.73
Test configuration:
GPU: RTX 4090, RTX 3090.
Windows 11: 23H2
NVIDIA Graphics driver: R565 or higher
Inference:
Inference Backend: Onnxruntime-GenAI-DirectML
We used GenAI ORT->DML backend for inference. The instructions to use this backend are given in readme.txt file available under Files section.
Ethical Considerations:
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.
Please report security vulnerabilities or NVIDIA AI Concerns here.