Update README.md
Browse files
README.md
CHANGED
|
@@ -12,7 +12,7 @@ base_model:
|
|
| 12 |
- **Input:** Text
|
| 13 |
- **Output:** Text
|
| 14 |
- **Supported Hardware Microarchitecture:** AMD MI350/MI355
|
| 15 |
-
- **ROCm**
|
| 16 |
- **Operating System(s):** Linux
|
| 17 |
- **Inference Engine:** [vLLM](https://docs.vllm.ai/en/latest/)
|
| 18 |
- **Model Optimizer:** [AMD-Quark](https://quark.docs.amd.com/latest/index.html)
|
|
@@ -77,6 +77,12 @@ The model was evaluated on GSM8K benchmarks.
|
|
| 77 |
### Reproduction
|
| 78 |
The GSM8K results were obtained using the [lm-evaluation-harness](https://github.com/EleutherAI/lm-evaluation-harness.git) framework, based on the Docker image `rocm/vllm-dev:base`, with vLLM and lm-eval compiled and installed from source inside the container.
|
| 79 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 80 |
#### Launching server
|
| 81 |
```
|
| 82 |
export VLLM_ATTENTION_BACKEND="TRITON_MLA"
|
|
|
|
| 12 |
- **Input:** Text
|
| 13 |
- **Output:** Text
|
| 14 |
- **Supported Hardware Microarchitecture:** AMD MI350/MI355
|
| 15 |
+
- **ROCm:** 7.0
|
| 16 |
- **Operating System(s):** Linux
|
| 17 |
- **Inference Engine:** [vLLM](https://docs.vllm.ai/en/latest/)
|
| 18 |
- **Model Optimizer:** [AMD-Quark](https://quark.docs.amd.com/latest/index.html)
|
|
|
|
| 77 |
### Reproduction
|
| 78 |
The GSM8K results were obtained using the [lm-evaluation-harness](https://github.com/EleutherAI/lm-evaluation-harness.git) framework, based on the Docker image `rocm/vllm-dev:base`, with vLLM and lm-eval compiled and installed from source inside the container.
|
| 79 |
|
| 80 |
+
#### Commit Hash:
|
| 81 |
+
- **vLLM:** [cbbae38f9368b6c35d9b9295bf4ceee1e6452750](https://github.com/vllm-project/vllm/commit/cbbae38f9368b6c35d9b9295bf4ceee1e6452750)
|
| 82 |
+
- **Quark:** [1742e8c40f8b90c4ecb2b086788160e919986399](https://gitenterprise.xilinx.com/AMDNeuralOpt/Quark/commit/1742e8c40f8b90c4ecb2b086788160e919986399)
|
| 83 |
+
- **lm-evaluation-harness:** [4b74ec1268267ea2ea83893400d7013df30507af](https://github.com/EleutherAI/lm-evaluation-harness/commit/4b74ec1268267ea2ea83893400d7013df30507af)
|
| 84 |
+
- tips: Remove the 'hf' and 'vllm' optional dependency lines from pyproject.toml in the lm-evaluation-harness repository to prevent installation errors.
|
| 85 |
+
|
| 86 |
#### Launching server
|
| 87 |
```
|
| 88 |
export VLLM_ATTENTION_BACKEND="TRITON_MLA"
|