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

Fix broken CI TestVILAServer (#7610)

parent ce3a3e87
......@@ -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"
......
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