Skip to content

Commit

Permalink
Fix wrong vllm repo. (#1454)
Browse files Browse the repository at this point in the history
Use vllm-fork for gaudi.

fix the issue #1451
  • Loading branch information
lkk12014402 authored Jan 22, 2025
1 parent ee0e5cc commit f07cf1d
Showing 1 changed file with 4 additions and 6 deletions.
10 changes: 4 additions & 6 deletions AgentQnA/tests/step1_build_images.sh
Original file line number Diff line number Diff line change
Expand Up @@ -38,13 +38,11 @@ function build_vllm_docker_image() {
echo "Building the vllm docker image"
cd $WORKPATH
echo $WORKPATH
if [ ! -d "./vllm" ]; then
echo "clone vllm repo...."
git clone https://github.com/vllm-project/vllm.git
if [ ! -d "./vllm-fork" ]; then
git clone https://github.com/HabanaAI/vllm-fork.git
fi
cd ./vllm
echo "Checking out latest stable release of vllm"
git checkout v0.6.6
cd ./vllm-fork
git checkout v0.6.4.post2+Gaudi-1.19.0
docker build --no-cache -f Dockerfile.hpu -t opea/vllm-gaudi:ci --shm-size=128g . --build-arg https_proxy=$https_proxy --build-arg http_proxy=$http_proxy
if [ $? -ne 0 ]; then
echo "opea/vllm-gaudi:ci failed"
Expand Down

0 comments on commit f07cf1d

Please sign in to comment.