Skip to content

Commit

Permalink
debug
Browse files Browse the repository at this point in the history
Signed-off-by: Sivanantham Chinnaiyan <[email protected]>
  • Loading branch information
sivanantha321 committed Jun 26, 2024
1 parent 7207360 commit 8194b73
Showing 1 changed file with 8 additions and 1 deletion.
9 changes: 8 additions & 1 deletion test/e2e/predictor/test_huggingface.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@

import pytest
from kubernetes import client
from kubernetes.client import V1ResourceRequirements
from kubernetes.client import V1ResourceRequirements, V1EnvVar

from kserve import (
V1beta1PredictorSpec,
Expand Down Expand Up @@ -302,6 +302,13 @@ def test_vllm_openai_chat_completions():
"--device",
"cpu",
],
env=[
V1EnvVar(
"VLLM_LOGGING_LEVEL",
"DEBUG",
),
V1EnvVar("VLLM_TRACE_FUNCTION", "1"),
], # Add environment variables
resources=V1ResourceRequirements(
requests={"cpu": "1", "memory": "2Gi"},
limits={"cpu": "1", "memory": "4Gi"},
Expand Down

0 comments on commit 8194b73

Please sign in to comment.