Skip to content

Commit d7374d7

Browse files
authored
Fix broken CI TestVILAServer (sgl-project#7610)
1 parent ce3a3e8 commit d7374d7

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

test/srt/test_vision_openai_server_b.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -225,9 +225,10 @@ def test_video_chat_completion(self):
225225
class TestVILAServer(TestOpenAIVisionServer):
226226
@classmethod
227227
def setUpClass(cls):
228-
cls.model = "AndyZijianZhang/NVILA-Lite-2B"
228+
cls.model = "Efficient-Large-Model/NVILA-Lite-2B-hf-0626"
229229
cls.base_url = DEFAULT_URL_FOR_TEST
230230
cls.api_key = "sk-123456"
231+
cls.revision = "6bde1de5964b40e61c802b375fff419edc867506"
231232
cls.process = popen_launch_server(
232233
cls.model,
233234
cls.base_url,
@@ -236,6 +237,7 @@ def setUpClass(cls):
236237
other_args=[
237238
"--trust-remote-code",
238239
"--context-length=65536",
240+
f"--revision={cls.revision}",
239241
],
240242
)
241243
cls.base_url += "/v1"

0 commit comments

Comments
 (0)