diff --git a/Dockerfile b/Dockerfile index de1c90cbd..e049bc166 100644 --- a/Dockerfile +++ b/Dockerfile @@ -15,7 +15,7 @@ RUN pip install -r requirements.txt EXPOSE 3306 EXPOSE 8000 -CMD ["python", "pilot/server/vicuna_server.py"] +CMD ["python", "pilot/server/llmserver.py"] CMD ["python", "pilot/server/webserver.py"] diff --git a/docker-compose.yml b/docker-compose.yml index fbf05f4c9..dca53ef2b 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -1,4 +1,4 @@ -version: '3.9' +version: '3.10' services: db-gpt: diff --git a/environment.yml b/environment.yml index 756f5f465..99959ec5f 100644 --- a/environment.yml +++ b/environment.yml @@ -4,7 +4,7 @@ channels: - defaults - anaconda dependencies: - - python=3.9 + - python=3.10 - cudatoolkit - pip - pytorch-mutex=1.0=cuda diff --git a/run.sh b/run.sh index 8af83357b..a8948ad0f 100644 --- a/run.sh +++ b/run.sh @@ -15,7 +15,7 @@ function find_python_command() { PYTHONCMD=$(find_python_command) -nohup PYTHONCMD pilot/server/vicuna_server.py >> /root/server.log 2>&1 & +nohup PYTHONCMD pilot/server/llmserver.py >> /root/server.log 2>&1 & while [ `grep -c "Uvicorn running on" /root/server.log` -eq '0' ];do sleep 1s; echo "wait server running"