The NVIDIA Gemma-2b-it ONNX INT4 model is the quantized version of the Google Gemma-2b-it model which is a text-to-text, decoder-only large language models, available in English, with open weights, pre-trained variants, and instruction-tuned variants. It is well-suited for a variety of text generation tasks, including question answering, summarization, and reasoning. Relatively small size makes it possible to deploy them in environments with limited resources such as a laptop, desktop or your own cloud infrastructure. For more information, please check
here. The NVIDIA Gemma-2b-it ONNX INT4 model is quantized with
TensorRT Model Optimizer.
This model is ready for commercial and research use case.
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
Gemma-2b-it Model Card.
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).
-
Input Type: Text
-
Input Format: String
-
Input Parameters: Sequence (1D)
-
Other Properties Related to Input: Text strings can include a question, prompt, or a document to be summarized. Primarily for English language
-
Output Type: Text
-
Output Format: String
-
Output Parameters: Sequence (1D)
-
Other properties: Generated English-language text in response to the input, such as an answer to a question, or a summary of a document.
-
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
-
Data Collection Method by dataset: Automated
-
Labeling Method by dataset: [Unknown]
-
Data Collection Method by dataset - Unknown
-
Labeling Method by dataset - Not Applicable
With GenAI ORT->DML backend, we got below mentioned accuracy numbers on a desktop RTX 4090 GPU system.
-
GPU: RTX 4090, RTX 3090.
-
Windows 11: 23H2
-
NVIDIA Graphics driver: R565 or higher
We used GenAI ORT->DML backend for inference. The instructions to use this backend are given in readme.txt file available under Files section.
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.
