modify vllm@f721096 for AlphaMath to support value model
October 24, 2024 ยท View on GitHub
Installation
- Method 1
git clone https://github.com/MARIO-Math-Reasoning/vllm
cd vllm
pip install -e .
- Method 2
- Install the original vllm v0.3.3
git clone https://github.com/vllm-project/vllm.git cd vllm git checkout f721096 pip install -e .- Copy the following modified files and added files from this repo to the above vllm folder.
modified files
vllm/core/scheduler.py
vllm/engine/llm_engine.py
vllm/model_executor/model_loader.py
vllm/model_executor/models/__init__.py
vllm/outputs.py
vllm/sequence.py
vllm/worker/model_runner.py
pyproject.toml # because setuptools 70.0.0 was released May 21 2024, need to pin version of setuptools.
requirements-build.txt # because setuptools 70.0.0 was released May 21 2024, need to pin version of setuptools.
added files
tests/test_regression_withvalue.py
vllm/model_executor/models/modeling_value_head.py