Skip to content

GitLab

Unverified Commit d7374d74 authored by Lifu Huang's avatar Lifu Huang Committed by GitHub
Browse files

Fix broken CI TestVILAServer (#7610)

parent ce3a3e87
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
test/srt/test_vision_openai_server_b.py
View file @ d7374d74
...@@ -225,9 +225,10 @@ class TestPhi4MMServer(TestOpenAIVisionServer): ...@@ -225,9 +225,10 @@ class TestPhi4MMServer(TestOpenAIVisionServer):
class TestVILAServer(TestOpenAIVisionServer): class TestVILAServer(TestOpenAIVisionServer):
@classmethod @classmethod
def setUpClass(cls): def setUpClass(cls):
cls.model = "AndyZijianZhang/NVILA-Lite-2B" cls.model = "Efficient-Large-Model/NVILA-Lite-2B-hf-0626"
cls.base_url = DEFAULT_URL_FOR_TEST cls.base_url = DEFAULT_URL_FOR_TEST
cls.api_key = "sk-123456" cls.api_key = "sk-123456"
cls.revision = "6bde1de5964b40e61c802b375fff419edc867506"
cls.process = popen_launch_server( cls.process = popen_launch_server(
cls.model, cls.model,
cls.base_url, cls.base_url,
...@@ -236,6 +237,7 @@ class TestVILAServer(TestOpenAIVisionServer): ...@@ -236,6 +237,7 @@ class TestVILAServer(TestOpenAIVisionServer):
other_args=[ other_args=[
"--trust-remote-code", "--trust-remote-code",
"--context-length=65536", "--context-length=65536",
f"--revision={cls.revision}",
], ],
) )
cls.base_url += "/v1" cls.base_url += "/v1"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment