diff --git a/services/vllm/config_bge.yaml b/services/vllm/config_bge.yaml index f1a8c15..f8cd65b 100644 --- a/services/vllm/config_bge.yaml +++ b/services/vllm/config_bge.yaml @@ -2,6 +2,6 @@ model: BAAI/bge-large-en-v1.5 host: "0.0.0.0" port: 8002 uvicorn-log-level: "info" -gpu-memory-utilization: 0.08 +gpu-memory-utilization: 0.06 dtype: float16 task: embed diff --git a/services/vllm/config_llama.yaml b/services/vllm/config_llama.yaml index de48363..9cfeb6b 100644 --- a/services/vllm/config_llama.yaml +++ b/services/vllm/config_llama.yaml @@ -2,7 +2,7 @@ model: meta-llama/Llama-3.1-8B-Instruct host: "0.0.0.0" port: 8001 uvicorn-log-level: "info" -gpu-memory-utilization: 0.80 +gpu-memory-utilization: 0.85 max-model-len: 24576 dtype: auto enforce-eager: false