nvidia/DeepSeek-R1-0528-NVFP4-v2

NVIDIA

The NVIDIA DeepSeek-R1-0528-FP4 v2 model is the quantized version of the DeepSeek AI's DeepSeek R1 0528 model, which is an auto-regressive language model that uses an optimized transformer architecture. For more information, please check [here](https://huggingface.co/deepseek-ai/DeepSeek-R1-0528). The NVIDIA DeepSeek R1 FP4 model is quantized with [TensorRT Model Optimizer](https://github.com/NVIDIA/TensorRT-Model-Optimizer). Compared to [nvidia/DeepSeek-R1-0528-FP4](https://huggingface.co/nvidia/DeepSeek-R1-0528-FP4), this checkpoint additionally quantizes the wo module in attention layers. This model is ready for commercial/non-commercial use. <br> 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 [(DeepSeek R1) Model Card](https://huggingface.co/deepseek-ai/DeepSeek-R1-0528). [MIT](https://huggingface.co/datasets/choosealicense/licenses/blob/main/markdown/mit.md) **Architecture Type:** Transformers <br> **Network Architecture:** DeepSeek R1 <br> **Input Type(s):** Text <br> **Input Format(s):** String <br> **Input Parameters:** 1D (One Dimensional): Sequences <br> **Other Properties Related to Input:** DeepSeek recommends adhering to the following configurations when utilizing the DeepSeek-R1 series models, including benchmarking, to achieve the expected performance: \ - Set the temperature within the range of 0.5-0.7 (0.6 is recommended) to prevent endless repetitions or incoherent outputs. - Avoid adding a system prompt; all instructions should be contained within the user prompt. - For mathematical problems, it is advisable to include a directive in your prompt such as: "Please reason step by step, and put your final answer within \boxed{}." - When evaluating model performance, it is recommended to conduct multiple tests and average the results. <br> **Output Type(s):** Text <br> **Output Format:** String <br> **Output Parameters:** 1D (One Dimensional): Sequences <br> **Supported Runtime Engine(s):** <br> * TensorRT-LLM <br> **Supported Hardware Microarchitecture Compatibility:** <br> * NVIDIA Blackwell <br> **Preferred Operating System(s):** <br> * Linux <br> ** The model is quantized with nvidia-modelopt **v0.33.0** <br> ** Data Collection Method by dataset: Hybrid: Human, Automated <br> ** Labeling Method by dataset: Hybrid: Human, Automated <br> ** Data Collection Method by dataset: Hybrid: Human, Automated <br> ** Labeling Method by dataset: Hybrid: Human, Automated <br> ** Data Collection Method by dataset: Hybrid: Human, Automated <br> ** Labeling Method by dataset: Hybrid: Human, Automated <br> * Calibration Dataset: [cnn_dailymail](https://huggingface.co/datasets/abisee/cnn_dailymail) <br> ** Data collection method: Automated. <br> ** Labeling method: Automated. <br> **Engine:** TensorRT-LLM <br> **Test Hardware:** B200 <br> This model was obtained by quantizing the weights and activations of DeepSeek R1-0528 to FP4 data type, ready for inference with TensorRT-LLM. Only the weights and activations of the linear operators within transformer blocks are quantized. This optimization reduces the number of bits per parameter from 8 to 4, reducing the disk size and GPU memory requirements by approximately 1.6x. To deploy the quantized FP4 checkpoint with [TensorRT-LLM](https://github.com/NVIDIA/TensorRT-LLM) LLM API, follow the sample codes below (you need 8xB200 GPU and TensorRT LLM built from source with the latest main branch): * LLM API sample usage: from tensorrt_llm import SamplingParams from tensorrt_llm._torch import LLM def main(): prompts = [ "Hello, my name is", "The president of the United States is", "The capital of France is", "The future of AI is", ] sampling_params = SamplingParams(max_tokens=32) llm = LLM(model="nvidia/DeepSeek-R1-0528-FP4-v2", tensor_parallel_size=8, enable_attention_dp=True) outputs = llm.generate(prompts, sampling_params) for output in outputs: prompt = output.prompt generated_text = output.outputs[0].text print(f"Prompt: {prompt!r}, Generated text: {generated_text!r}") if __name__ == '__main__': main() The accuracy benchmark results are presented in the table below: *Max OSL for LiveCodeBench eval can be as high as 128K. The base model was trained on data that contains toxic language and societal biases originally crawled from the internet. Therefore, the model may amplify those biases and return toxic responses especially when prompted with toxic prompts. The model may generate answers that may be inaccurate, omit key information, or include irrelevant or redundant text producing socially unacceptable or undesirable text, even if the prompt itself does not include anything explicitly offensive. 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](https://www.nvidia.com/en-us/support/submit-security-vulnerability/).