Fix broken CI TestVILAServer (#7610)

This commit is contained in:
Lifu Huang
2025-06-27 15:01:03 -07:00
committed by GitHub
parent ce3a3e8783
commit d7374d7467

View File

@@ -225,9 +225,10 @@ class TestPhi4MMServer(TestOpenAIVisionServer):
class TestVILAServer(TestOpenAIVisionServer):
@classmethod
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.api_key = "sk-123456"
cls.revision = "6bde1de5964b40e61c802b375fff419edc867506"
cls.process = popen_launch_server(
cls.model,
cls.base_url,
@@ -236,6 +237,7 @@ class TestVILAServer(TestOpenAIVisionServer):
other_args=[
"--trust-remote-code",
"--context-length=65536",
f"--revision={cls.revision}",
],
)
cls.base_url += "/v1"