Update README.md
Browse files
README.md
CHANGED
|
@@ -52,11 +52,6 @@ model-index:
|
|
| 52 |
|
| 53 |
This model is a quantized version of OpenAI's GPT-OSS-20B using NVIDIA's advanced NVFP4 format. It follows the official NVIDIA TensorRT Model Optimizer methodology, providing superior accuracy retention compared to MXFP4 quantization while maintaining significant memory efficiency gains.
|
| 54 |
|
| 55 |
-
## Blog
|
| 56 |
-
Fine-Tuning gpt-oss for Accuracy and Performance with Quantization Aware Training:
|
| 57 |
-
https://developer.nvidia.com/blog/fine-tuning-gpt-oss-for-accuracy-and-performance-with-quantization-aware-training/
|
| 58 |
-
|
| 59 |
-
|
| 60 |
## Key Features
|
| 61 |
|
| 62 |
- **Advanced Quantization**: Uses NVFP4 format with FP8 E4M3 scaling for enhanced precision
|
|
@@ -207,6 +202,12 @@ This approach can achieve up to 98% task-specific performance recovery.
|
|
| 207 |
- **Memory**: 24GB+ VRAM recommended
|
| 208 |
- **CUDA**: 12.0+
|
| 209 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 210 |
### Framework Support Status
|
| 211 |
|
| 212 |
- **TensorRT-LLM**: NVFP4 support in active development
|
|
@@ -216,10 +217,13 @@ This approach can achieve up to 98% task-specific performance recovery.
|
|
| 216 |
|
| 217 |
## Model Format Details
|
| 218 |
|
| 219 |
-
- **Storage Format**:
|
| 220 |
-
- **
|
| 221 |
-
- **
|
| 222 |
-
- **Size**: ~
|
|
|
|
|
|
|
|
|
|
| 223 |
|
| 224 |
## Use Cases
|
| 225 |
|
|
@@ -235,6 +239,23 @@ This approach can achieve up to 98% task-specific performance recovery.
|
|
| 235 |
- **Memory**: ~75% reduction in deployment memory requirements
|
| 236 |
- **Compatibility**: Works with standard transformers, optimized for NVIDIA frameworks
|
| 237 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 238 |
## License
|
| 239 |
|
| 240 |
This model inherits the Apache 2.0 license from the base openai/gpt-oss-20b model. Commercial use is permitted under the same terms.
|
|
|
|
| 52 |
|
| 53 |
This model is a quantized version of OpenAI's GPT-OSS-20B using NVIDIA's advanced NVFP4 format. It follows the official NVIDIA TensorRT Model Optimizer methodology, providing superior accuracy retention compared to MXFP4 quantization while maintaining significant memory efficiency gains.
|
| 54 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 55 |
## Key Features
|
| 56 |
|
| 57 |
- **Advanced Quantization**: Uses NVFP4 format with FP8 E4M3 scaling for enhanced precision
|
|
|
|
| 202 |
- **Memory**: 24GB+ VRAM recommended
|
| 203 |
- **CUDA**: 12.0+
|
| 204 |
|
| 205 |
+
### Compatible Hardware (Software Emulation)
|
| 206 |
+
- **RTX 4090**: Ada Lovelace architecture (no native NVFP4 acceleration)
|
| 207 |
+
- **RTX 4080/4070**: Compatible via software emulation
|
| 208 |
+
- **Data Center**: H100, A100 (software emulation)
|
| 209 |
+
- **Memory**: 20GB+ VRAM for model loading
|
| 210 |
+
|
| 211 |
### Framework Support Status
|
| 212 |
|
| 213 |
- **TensorRT-LLM**: NVFP4 support in active development
|
|
|
|
| 217 |
|
| 218 |
## Model Format Details
|
| 219 |
|
| 220 |
+
- **Storage Format**: BF16 with NVFP4 quantization metadata
|
| 221 |
+
- **File Size**: ~39GB (BF16 precision with quantization instructions)
|
| 222 |
+
- **Deployment Format**: Runtime conversion to NVFP4 by compatible inference engines
|
| 223 |
+
- **Deployed Size**: ~10GB when converted to 4-bit NVFP4 format
|
| 224 |
+
- **File Format**: SafeTensors with embedded quantization configuration
|
| 225 |
+
|
| 226 |
+
This model contains the full BF16 weights along with quantization parameters that enable inference engines like TensorRT-LLM to convert weights to true 4-bit NVFP4 format during model loading. The memory savings and performance benefits are realized at inference time, not during storage.
|
| 227 |
|
| 228 |
## Use Cases
|
| 229 |
|
|
|
|
| 239 |
- **Memory**: ~75% reduction in deployment memory requirements
|
| 240 |
- **Compatibility**: Works with standard transformers, optimized for NVIDIA frameworks
|
| 241 |
|
| 242 |
+
## Limitations and Considerations
|
| 243 |
+
|
| 244 |
+
- **Current State**: Model saved in fake-quantized format for compatibility
|
| 245 |
+
- **Real Benefits**: Achieved only when deployed with NVFP4-compatible engines
|
| 246 |
+
- **Hardware Dependency**: Optimal performance requires NVIDIA Blackwell architecture
|
| 247 |
+
- **Framework Support**: Limited until inference engines implement NVFP4 support
|
| 248 |
+
- **Model Size**: Large storage footprint until deployment conversion
|
| 249 |
+
|
| 250 |
+
## Evaluation and Benchmarking
|
| 251 |
+
|
| 252 |
+
This model maintains the capabilities of the original GPT-OSS-20B while providing memory efficiency benefits. For comprehensive evaluation, test against:
|
| 253 |
+
|
| 254 |
+
- **Language Modeling**: Perplexity on standard datasets
|
| 255 |
+
- **Downstream Tasks**: Task-specific accuracy measurements
|
| 256 |
+
- **Generation Quality**: Human evaluation of output coherence
|
| 257 |
+
- **Memory Usage**: Deployment memory requirements vs. accuracy trade-offs
|
| 258 |
+
|
| 259 |
## License
|
| 260 |
|
| 261 |
This model inherits the Apache 2.0 license from the base openai/gpt-oss-20b model. Commercial use is permitted under the same terms.
|