Skip to content

Commit

Permalink
fix: disable logging vllm requests
Browse files Browse the repository at this point in the history
  • Loading branch information
Avram Tudor committed Oct 8, 2024
1 parent 760aa16 commit 6b5a4c7
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 1 deletion.
2 changes: 1 addition & 1 deletion run.sh
Original file line number Diff line number Diff line change
Expand Up @@ -9,5 +9,5 @@ else
cd ..
fi

export LLAMA_N_CTX=8192
export LLAMA_N_CTX=16000
poetry run python -m uvicorn skynet.main:app --reload
1 change: 1 addition & 0 deletions skynet/modules/ttt/openai_api/app.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,7 @@ def initialize():
openai_api_server_path = vllm_server_path
proc = subprocess.Popen(
f'python -m {openai_api_server_path} \
--disable-log-requests \
--model {llama_path} \
--gpu_memory_utilization 0.95 \
--max-model-len {llama_n_ctx} \
Expand Down

0 comments on commit 6b5a4c7

Please sign in to comment.