test : fix multi-GPU server tests (#27614)

* tests : fix tests for multi-gpu environment

* cont : not needed
This commit is contained in:
Georgi Gerganov
2026-08-23 19:59:42 +03:00
committed by GitHub
parent 4a08fa2970
commit 8d9af25633
3 changed files with 22 additions and 3 deletions
+1 -1
View File
@@ -121,7 +121,7 @@ def test_vision_chat_completion_token_count():
"prompt, image_data, success, re_content",
[
# test model is trained on CIFAR-10, but it's quite dumb due to small size
("What is this: <__media__>\n", "IMG_BASE64_0", True, "(cat)+"),
("What is this: <__media__>\n", "IMG_BASE64_0", True, "(cat)+|(automobile)+"),
("What is this: <__media__>\n", "IMG_BASE64_1", True, "(frog)+"),
("What is this: <__media__>\n", "malformed", False, None), # non-image data
("What is this:\n", "", False, None), # empty string