Skip to content

Commit 7a20ebe

Browse files
committed
Reduce test token length
1 parent 62322ba commit 7a20ebe

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

tests/functional_tests/tests_inframework/test_deploy_query_vlm_ray.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,7 @@ def test_deploy_ray(self):
103103
port=8000,
104104
model_id="megatron-multimodal",
105105
prompt="What is the color of a banana?",
106-
max_tokens=20,
106+
max_tokens=5,
107107
)
108108

109109
print(output)
@@ -123,7 +123,7 @@ def test_deploy_ray(self):
123123
port=8000,
124124
model_id="megatron-multimodal",
125125
prompt=chat_messages,
126-
max_tokens=20,
126+
max_tokens=5,
127127
use_chat=True,
128128
)
129129
print(output_chat)
@@ -148,7 +148,7 @@ def test_deploy_ray(self):
148148
port=8000,
149149
model_id="megatron-multimodal",
150150
messages=messages_with_image,
151-
max_tokens=16,
151+
max_tokens=5,
152152
)
153153
print(output_image)
154154
assert output_image != "", "Chat with base64 image returned empty"

0 commit comments

Comments
 (0)