ONNX Model Zoo

RedHatAI/mpt-7b-gsm8k-quant-ds

By RedHatAI · ↓ 3 · ♥ 1 · Updated 2023-10-18

What's in this repo

This page lists the files in RedHatAI/mpt-7b-gsm8k-quant-ds, including model-orig.onnx, model.onnx.

  • 2 ONNX models (.onnx files)
  • 9 files, including 4 .json, 2 .onnx, 1 .md, 1 .data
  • Last updated 2023-10-18

ONNX files

File Size Download
model-orig.onnx — Download
model.onnx — Download

How to use

These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.

README

mpt-7b-gsm8k

Paper: Sparse Finetuning for Inference Acceleration of Large Language Models
Code: https://github.com/neuralmagic/deepsparse/tree/main/research/mpt

This model was produced from a MPT-7B base model finetuned on the GSM8k dataset for 2 epochs and quantized.

GSM8k zero-shot accuracy with lm-evaluation-harness : 27.8%

All MPT model weights are available on SparseZoo and CPU speedup for generative inference can be reproduced by following the instructions at DeepSparse

Model Links Compression
neuralmagic/mpt-7b-gsm8k-quant Quantization (W8A8)
neuralmagic/mpt-7b-gsm8k-pruned40-quant Quantization (W8A8) & 40% Pruning
neuralmagic/mpt-7b-gsm8k-pruned50-quant Quantization (W8A8) & 50% Pruning
neuralmagic/mpt-7b-gsm8k-pruned60-quant Quantization (W8A8) & 60% Pruning
neuralmagic/mpt-7b-gsm8k-pruned70-quant Quantization (W8A8) & 70% Pruning
neuralmagic/mpt-7b-gsm8k-pruned70-quant Quantization (W8A8) & 75% Pruning
neuralmagic/mpt-7b-gsm8k-pruned80-quant Quantization (W8A8) & 80% Pruning

For general questions on these models and sparsification methods, reach out to the engineering team on our community Slack.

Model info

text-generation transformers
transformers onnx mpt text-generation deepsparse custom_code dataset:gsm8k arxiv:2310.06927 region:us
↓ 3 ♥ 1

Created: 2023-10-10

Updated: 2023-10-18

View on Hugging Face

Files (9)

.gitattributes
README.md
config.json
model-orig.onnx ONNX
model.data
model.onnx ONNX
special_tokens_map.json
tokenizer.json
tokenizer_config.json

Related models