Skip to content

MARIO-Math-Reasoning/vllm

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

modify vllm@f721096 for AlphaMath to support value model

Installation

  • Method 1
git clone https://github.com/MARIO-Math-Reasoning/vllm
cd vllm
pip install -e .
  • Method 2
    1. Install the original vllm v0.3.3
    git clone https://github.com/vllm-project/vllm.git
    cd vllm
    git checkout f721096
    pip install -e .
    1. 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