We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 62322ba commit 7a20ebeCopy full SHA for 7a20ebe
1 file changed
tests/functional_tests/tests_inframework/test_deploy_query_vlm_ray.py
@@ -103,7 +103,7 @@ def test_deploy_ray(self):
103
port=8000,
104
model_id="megatron-multimodal",
105
prompt="What is the color of a banana?",
106
- max_tokens=20,
+ max_tokens=5,
107
)
108
109
print(output)
@@ -123,7 +123,7 @@ def test_deploy_ray(self):
123
124
125
prompt=chat_messages,
126
127
use_chat=True,
128
129
print(output_chat)
@@ -148,7 +148,7 @@ def test_deploy_ray(self):
148
149
150
messages=messages_with_image,
151
- max_tokens=16,
152
153
print(output_image)
154
assert output_image != "", "Chat with base64 image returned empty"
0 commit comments