初始化项目,由ModelHub XC社区提供模型
Model: neuralmagic/Sparse-Llama-3.1-8B-evolcodealpaca-2of4-FP8-dynamic Source: Original Platform
This commit is contained in:
77
README.md
Normal file
77
README.md
Normal file
@@ -0,0 +1,77 @@
|
||||
---
|
||||
tags:
|
||||
- vllm
|
||||
- sparsity
|
||||
- quantization
|
||||
- int4
|
||||
pipeline_tag: text-generation
|
||||
license: llama3.1
|
||||
base_model: neuralmagic/Sparse-Llama-3.1-8B-evolcodealpaca-2of4
|
||||
datasets:
|
||||
- theblackcat102/evol-codealpaca-v1
|
||||
language:
|
||||
- en
|
||||
---
|
||||
|
||||
# Sparse-Llama-3.1-8B-evolcodealpaca-2of4-FP8-dynamic
|
||||
|
||||
## Model Overview
|
||||
- **Model Architecture:** Llama-3.1-8B
|
||||
- **Input:** Text
|
||||
- **Output:** Text
|
||||
- **Model Optimizations:**
|
||||
- **Sparsity:** 2:4
|
||||
- **Weight quantization:** FP8
|
||||
- **Activation quantization:** FP8
|
||||
- **Release Date:** 11/15/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 a code completion AI model obtained by fine-tuning the 2:4 sparse [Sparse-Llama-3.1-8B-2of4](https://huggingface.co/neuralmagic/Sparse-Llama-3.1-8B-2of4) on the [evol-codealpaca-v1](https://huggingface.co/datasets/theblackcat102/evol-codealpaca-v1) dataset, followed by quantization
|
||||
On the [HumanEval](https://arxiv.org/abs/2107.03374) benchmark, it achieves a pass@1 of 49.0, compared to 48.5 for the fine-tuned dense model [Llama-3.1-8B-evolcodealpaca](https://huggingface.co/neuralmagic/Llama-3.1-8B-evolcodealpaca) — demonstrating over **100% accuracy recovery**.
|
||||
|
||||
|
||||
### Model Optimizations
|
||||
|
||||
This model was obtained by quantizing the weights and of [Sparse-Llama-3.1-8B-evolcodealpaca-2of4](https://huggingface.co/neuralmagic/Sparse-Llama-3.1-8B-evolcodealpaca-2of4) to FP8 data type.
|
||||
This optimization reduces the number of bits used to represent weights and activations from 16 to 8, reducing GPU memory requirements (by approximately 50%) and increasing matrix-multiply compute throughput (by approximately 2x).
|
||||
Weight quantization also reduces disk size requirements by approximately 50%.
|
||||
|
||||
Only weights and activations of the linear operators within transformers blocks are quantized.
|
||||
Weights are quantized with a symmetric static per-channel scheme, where a fixed linear scaling factor is applied between FP8 and BF16 representations for each output channel dimension.
|
||||
Linear scaling factors are computed via by minimizing the mean squarred error (MSE).
|
||||
Activations are quantized with a symmetric dynamic per-token scheme, computing a linear scaling factor at runtime for each token between FP8 and BF16 representations.
|
||||
|
||||
|
||||
## Deployment with vLLM
|
||||
|
||||
This model can be deployed efficiently using the [vLLM](https://docs.vllm.ai/en/latest/) backend. vLLM also supports OpenAI-compatible serving. See the [documentation](https://docs.vllm.ai/en/latest/) for more details.
|
||||
|
||||
|
||||
## Evaluation
|
||||
|
||||
This model was evaluated on Neural Magic's fork of [EvalPlus](https://github.com/neuralmagic/evalplus).
|
||||
|
||||
### Accuracy
|
||||
#### Human Benchmark
|
||||
<table>
|
||||
<tr>
|
||||
<td><strong>Metric</strong></td>
|
||||
<td style="text-align: center"><strong>Llama-3.1-8B-evolcodealpaca</strong></td>
|
||||
<td style="text-align: center"><strong>Sparse-Llama-3.1-8B-evolcodealpaca-2of4</strong></td>
|
||||
<td style="text-align: center"><strong>Sparse-Llama-3.1-8B-evolcodealpaca-2of4-FP8-dynamic</strong></td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td>HumanEval pass@1</td>
|
||||
<td style="text-align: center">48.5</td>
|
||||
<td style="text-align: center">49.1</td>
|
||||
<td style="text-align: center">49.0</td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td>HumanEval+ pass@1</td>
|
||||
<td style="text-align: center">44.2</td>
|
||||
<td style="text-align: center">46.3</td>
|
||||
<td style="text-align: center">46.2</td>
|
||||
</tr>
|
||||
</table>
|
||||
Reference in New Issue
Block a user