初始化项目,由ModelHub XC社区提供模型
Model: neuralmagic/Sparse-Llama-3.1-8B-2of4 Source: Original Platform
This commit is contained in:
36
.gitattributes
vendored
Normal file
36
.gitattributes
vendored
Normal file
@@ -0,0 +1,36 @@
|
|||||||
|
*.7z filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.arrow filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.bin filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.ckpt filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.ftz filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.gz filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.h5 filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.joblib filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.lfs.* filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.mlmodel filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.model filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.msgpack filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.npy filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.npz filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.onnx filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.ot filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.parquet filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.pb filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.pickle filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.pkl filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.pt filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.pth filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.rar filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.safetensors filter=lfs diff=lfs merge=lfs -text
|
||||||
|
saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.tar.* filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.tar filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.tflite filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.tgz filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.wasm filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.xz filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.zip filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.zst filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.json filter=lfs diff=lfs merge=lfs -text
|
||||||
148
README.md
Normal file
148
README.md
Normal file
@@ -0,0 +1,148 @@
|
|||||||
|
---
|
||||||
|
tags:
|
||||||
|
- vllm
|
||||||
|
- sparsity
|
||||||
|
pipeline_tag: text-generation
|
||||||
|
license: llama3.1
|
||||||
|
base_model: meta-llama/Llama-3.1-8B
|
||||||
|
---
|
||||||
|
|
||||||
|
# Get Started
|
||||||
|
Sparse-Llama-3.1 models use 2:4 semi-structured sparsity to deliver 2x model size and compute reduction.
|
||||||
|
Explore the [launch blog](https://neuralmagic.com/blog/24-sparse-llama-smaller-models-for-efficient-gpu-inference/) to learn more about Sparse-Llama-3.1 and its potential for efficient, scalable AI deployments.
|
||||||
|
You can also find all available models in our [Neural Magic HuggingFace collection](https://huggingface.co/collections/neuralmagic/sparse-llama-31-2of4-673f6e96ae74efa213cf1cff).
|
||||||
|
|
||||||
|
**Looking to build on top of sparse models?** Whether you aim to reduce deployment costs, improve inference performance, or create highly optimized versions for your enterprise needs, Sparse Llama provides the ideal foundation. These models offer state-of-the-art efficiency with 2:4 structured sparsity, enabling cost-effective scaling without sacrificing accuracy.
|
||||||
|
[Connect with us](https://neuralmagic.com/book-a-demo/) to explore how we can help integrate sparsity into your AI workflows.
|
||||||
|
|
||||||
|
|
||||||
|
# Sparse-Llama-3.1-8B-2of4
|
||||||
|
|
||||||
|
## Model Overview
|
||||||
|
- **Model Architecture:** Llama-3.1-8B
|
||||||
|
- **Input:** Text
|
||||||
|
- **Output:** Text
|
||||||
|
- **Model Optimizations:**
|
||||||
|
- **Sparsity:** 2:4
|
||||||
|
- **Release Date:** 11/20/2024
|
||||||
|
- **Version:** 1.0
|
||||||
|
- **License(s):** [llama3.1](https://huggingface.co/meta-llama/Meta-Llama-3.1-8B/blob/main/LICENSE)
|
||||||
|
- **Model Developers:** Neural Magic
|
||||||
|
|
||||||
|
This is the 2:4 sparse version of [Llama-3.1-8B](https://huggingface.co/meta-llama/Llama-3.1-8B).
|
||||||
|
On the [OpenLLM](https://huggingface.co/spaces/open-llm-leaderboard/open_llm_leaderboard) benchmark (version 1), it achieves an average score of 62.16, compared to 63.19 for the dense model—demonstrating a **98.37% accuracy recovery**. On the [Mosaic Eval Gauntlet](https://github.com/mosaicml/llm-foundry/blob/main/scripts/eval/local_data/EVAL_GAUNTLET.md) benchmark (version v0.3), it achieves an average score of 53.85, versus 55.34 for the dense model—representing a **97.3% accuracy recovery**.
|
||||||
|
|
||||||
|
|
||||||
|
### Model Optimizations
|
||||||
|
|
||||||
|
This model was obtained by pruning all linear operators within transformer blocks to the 2:4 sparsity pattern: in each group of four weights, two are retained while two are pruned. In addition to pruning, the sparse model was trained with knowledge distillation for 13B tokens to recover the accuracy loss incurred by pruning. For pruning, we utilize optimized version of [SparseGPT](https://arxiv.org/abs/2301.00774) through [LLM-Compressor](https://github.com/vllm-project/llm-compressor), and for sparse training with knowledge distillation we utilize [SquareHead approach](https://arxiv.org/abs/2310.06927).
|
||||||
|
|
||||||
|
|
||||||
|
## Deployment with vLLM
|
||||||
|
|
||||||
|
This model can be deployed efficiently using the [vLLM](https://docs.vllm.ai/en/latest/) backend. vLLM aslo supports OpenAI-compatible serving. See the [documentation](https://docs.vllm.ai/en/latest/) for more details.
|
||||||
|
|
||||||
|
|
||||||
|
## Evaluation
|
||||||
|
|
||||||
|
This model was evaluated on the [OpenLLM](https://huggingface.co/spaces/open-llm-leaderboard/open_llm_leaderboard) benchmark (version 1) with the [vLLM](https://docs.vllm.ai/en/stable/) engine for faster inference. In addition to the OpenLLM benchmark, the model was evaluated on the [Mosaic Eval Gauntlet](https://github.com/mosaicml/llm-foundry/blob/main/scripts/eval/local_data/EVAL_GAUNTLET.md) benchmark (version v0.3). The evaluation results are summarized below.
|
||||||
|
|
||||||
|
|
||||||
|
### Accuracy
|
||||||
|
|
||||||
|
#### Open LLM Leaderboard evaluation scores
|
||||||
|
|
||||||
|
|
||||||
|
<table>
|
||||||
|
<tr>
|
||||||
|
<td><strong>Benchmark</strong></td>
|
||||||
|
<td style="text-align: center"><strong>Llama-3.1-8B</strong></td>
|
||||||
|
<td style="text-align: center"><strong>Sparse-Llama-3.1-8B-2of4</strong></td>
|
||||||
|
</tr>
|
||||||
|
<tr>
|
||||||
|
<td>ARC-C (25-shot)</td>
|
||||||
|
<td style="text-align: center">58.2</td>
|
||||||
|
<td style="text-align: center">59.4</td>
|
||||||
|
</tr>
|
||||||
|
<tr>
|
||||||
|
<td>MMLU (5-shot)</td>
|
||||||
|
<td style="text-align: center">65.4</td>
|
||||||
|
<td style="text-align: center">60.6</td>
|
||||||
|
</tr>
|
||||||
|
<tr>
|
||||||
|
<td>HellaSwag (10-shot)</td>
|
||||||
|
<td style="text-align: center">82.3</td>
|
||||||
|
<td style="text-align: center">79.8</td>
|
||||||
|
</tr>
|
||||||
|
<tr>
|
||||||
|
<td>WinoGrande (5-shot)</td>
|
||||||
|
<td style="text-align: center">78.3</td>
|
||||||
|
<td style="text-align: center">75.9</td>
|
||||||
|
</tr>
|
||||||
|
<tr>
|
||||||
|
<td>GSM8K (5-shot)</td>
|
||||||
|
<td style="text-align: center">50.7</td>
|
||||||
|
<td style="text-align: center">56.3</td>
|
||||||
|
</tr>
|
||||||
|
<tr>
|
||||||
|
<td>TruthfulQA (0-shot)</td>
|
||||||
|
<td style="text-align: center">44.2</td>
|
||||||
|
<td style="text-align: center">40.9</td>
|
||||||
|
</tr>
|
||||||
|
<tr>
|
||||||
|
<td><strong>Average Score</strong></td>
|
||||||
|
<td style="text-align: center"><strong>63.19</strong></td>
|
||||||
|
<td style="text-align: center"><strong>62.16</strong></td>
|
||||||
|
</tr>
|
||||||
|
<tr>
|
||||||
|
<td><strong>Accuracy Recovery (%)</strong></td>
|
||||||
|
<td style="text-align: center"><strong>100</strong></td>
|
||||||
|
<td style="text-align: center"><strong>98.37</strong></td>
|
||||||
|
</tr>
|
||||||
|
</table>
|
||||||
|
|
||||||
|
|
||||||
|
#### Mosaic Eval Gauntlet evaluation scores
|
||||||
|
|
||||||
|
<table>
|
||||||
|
<tr>
|
||||||
|
<td><strong>Benchmark</strong></td>
|
||||||
|
<td style="text-align: center"><strong>Llama-3.1-8B</strong></td>
|
||||||
|
<td style="text-align: center"><strong>Sparse-Llama-3.1-8B-2of4</strong></td>
|
||||||
|
</tr>
|
||||||
|
<tr>
|
||||||
|
<td>World Knowledge</td>
|
||||||
|
<td style="text-align: center">59.4</td>
|
||||||
|
<td style="text-align: center">55.6</td>
|
||||||
|
</tr>
|
||||||
|
<tr>
|
||||||
|
<td>Commonsense Reasoning</td>
|
||||||
|
<td style="text-align: center">49.3</td>
|
||||||
|
<td style="text-align: center">50.0</td>
|
||||||
|
</tr>
|
||||||
|
<tr>
|
||||||
|
<td>Language Understanding</td>
|
||||||
|
<td style="text-align: center">69.8</td>
|
||||||
|
<td style="text-align: center">69.0</td>
|
||||||
|
</tr>
|
||||||
|
<tr>
|
||||||
|
<td>Symbolic Problem Solving</td>
|
||||||
|
<td style="text-align: center">40.0</td>
|
||||||
|
<td style="text-align: center">37.1</td>
|
||||||
|
</tr>
|
||||||
|
<tr>
|
||||||
|
<td>Reading Comprehension</td>
|
||||||
|
<td style="text-align: center">58.2</td>
|
||||||
|
<td style="text-align: center">57.5</td>
|
||||||
|
</tr>
|
||||||
|
<tr>
|
||||||
|
<td><strong>Average Score</strong></td>
|
||||||
|
<td style="text-align: center"><strong>55.34</strong></td>
|
||||||
|
<td style="text-align: center"><strong>53.85</strong></td>
|
||||||
|
</tr>
|
||||||
|
<tr>
|
||||||
|
<td><strong>Accuracy Recovery (%)</strong></td>
|
||||||
|
<td style="text-align: center"><strong>100</strong></td>
|
||||||
|
<td style="text-align: center"><strong>97.3</strong></td>
|
||||||
|
</tr>
|
||||||
|
</table>
|
||||||
3
config.json
Normal file
3
config.json
Normal file
@@ -0,0 +1,3 @@
|
|||||||
|
version https://git-lfs.github.com/spec/v1
|
||||||
|
oid sha256:4a911301d4d458d873929c8c64751889c3e86b203ed55369fcca5ce5284aed45
|
||||||
|
size 892
|
||||||
3
configuration.json
Normal file
3
configuration.json
Normal file
@@ -0,0 +1,3 @@
|
|||||||
|
version https://git-lfs.github.com/spec/v1
|
||||||
|
oid sha256:f888421726665e8a84b738eed42a64875aed79de8be7daade851ac8bf4c0cef9
|
||||||
|
size 73
|
||||||
3
generation_config.json
Normal file
3
generation_config.json
Normal file
@@ -0,0 +1,3 @@
|
|||||||
|
version https://git-lfs.github.com/spec/v1
|
||||||
|
oid sha256:f0f01e92153d8d962b8abe6e5ec47c258f2660fc24fc3aa1ea6bdd4b33b2a4df
|
||||||
|
size 185
|
||||||
3
model-00001-of-00004.safetensors
Normal file
3
model-00001-of-00004.safetensors
Normal file
@@ -0,0 +1,3 @@
|
|||||||
|
version https://git-lfs.github.com/spec/v1
|
||||||
|
oid sha256:49d2281bc1db84a9bb86058e9c5c43bb32dd47100ead793b14ffd934b4b00bad
|
||||||
|
size 4976698672
|
||||||
3
model-00002-of-00004.safetensors
Normal file
3
model-00002-of-00004.safetensors
Normal file
@@ -0,0 +1,3 @@
|
|||||||
|
version https://git-lfs.github.com/spec/v1
|
||||||
|
oid sha256:a5d45d17ed028bb1d3d4db6f456c4a1e036de68c54bedc5d7a462b8d23e13bfc
|
||||||
|
size 4999802720
|
||||||
3
model-00003-of-00004.safetensors
Normal file
3
model-00003-of-00004.safetensors
Normal file
@@ -0,0 +1,3 @@
|
|||||||
|
version https://git-lfs.github.com/spec/v1
|
||||||
|
oid sha256:a84d4a5b268728922cabb1802b08a58ac98c0fcbb43864af79f7a61edee34794
|
||||||
|
size 4915916176
|
||||||
3
model-00004-of-00004.safetensors
Normal file
3
model-00004-of-00004.safetensors
Normal file
@@ -0,0 +1,3 @@
|
|||||||
|
version https://git-lfs.github.com/spec/v1
|
||||||
|
oid sha256:d19475b36bc9b8219311797514104976a8e580a7ee5220b38caef1721b859c29
|
||||||
|
size 1168138808
|
||||||
3
model.safetensors.index.json
Normal file
3
model.safetensors.index.json
Normal file
@@ -0,0 +1,3 @@
|
|||||||
|
version https://git-lfs.github.com/spec/v1
|
||||||
|
oid sha256:146776fce3f6db1103aa6f249e65ee5544c5923ce6f971b092eee79aa6e5d37b
|
||||||
|
size 23950
|
||||||
3
special_tokens_map.json
Normal file
3
special_tokens_map.json
Normal file
@@ -0,0 +1,3 @@
|
|||||||
|
version https://git-lfs.github.com/spec/v1
|
||||||
|
oid sha256:cc2e013b7545f183ef03e079a3c91c6f364fa37e4068c512d7dd843e59024535
|
||||||
|
size 301
|
||||||
3
tokenizer.json
Normal file
3
tokenizer.json
Normal file
@@ -0,0 +1,3 @@
|
|||||||
|
version https://git-lfs.github.com/spec/v1
|
||||||
|
oid sha256:79e3e522635f3171300913bb421464a87de6222182a0570b9b2ccba2a964b2b4
|
||||||
|
size 9085657
|
||||||
3
tokenizer_config.json
Normal file
3
tokenizer_config.json
Normal file
@@ -0,0 +1,3 @@
|
|||||||
|
version https://git-lfs.github.com/spec/v1
|
||||||
|
oid sha256:4b4f36546c6125dd40594792627b965fb5a515688a0ecdaa0afd0b5f3ffcdba3
|
||||||
|
size 50498
|
||||||
Reference in New Issue
Block a user