Update README.md
Browse files
README.md
CHANGED
|
@@ -77,17 +77,14 @@ The model was evaluated on gsm8k benchmarks using the [vllm](https://github.com/
|
|
| 77 |
|
| 78 |
### Reproduction
|
| 79 |
|
| 80 |
-
The GSM8K results were obtained using the vLLM framework, based on the Docker image `rocm/vllm:
|
| 81 |
|
| 82 |
#### Preparation in container
|
| 83 |
```
|
| 84 |
-
#
|
| 85 |
-
pip uninstall vllm -y
|
| 86 |
git clone https://github.com/vllm-project/vllm.git
|
| 87 |
cd vllm
|
| 88 |
git checkout v0.13.0
|
| 89 |
-
pip install -r requirements/rocm.txt
|
| 90 |
-
python setup.py develop
|
| 91 |
cd ..
|
| 92 |
```
|
| 93 |
|
|
|
|
| 77 |
|
| 78 |
### Reproduction
|
| 79 |
|
| 80 |
+
The GSM8K results were obtained using the vLLM framework, based on the Docker image `rocm/vllm-dev:nightly`, and vLLM is installed inside the container.
|
| 81 |
|
| 82 |
#### Preparation in container
|
| 83 |
```
|
| 84 |
+
# Install vLLM code repo
|
|
|
|
| 85 |
git clone https://github.com/vllm-project/vllm.git
|
| 86 |
cd vllm
|
| 87 |
git checkout v0.13.0
|
|
|
|
|
|
|
| 88 |
cd ..
|
| 89 |
```
|
| 90 |
|