From d7374d74675e9d338424eb376418c775d0f54a2c Mon Sep 17 00:00:00 2001
From: Lifu Huang <lifu.hlf@gmail.com>
Date: Fri, 27 Jun 2025 15:01:03 -0700
Subject: [PATCH] Fix broken CI TestVILAServer (#7610)

---
 test/srt/test_vision_openai_server_b.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/test/srt/test_vision_openai_server_b.py b/test/srt/test_vision_openai_server_b.py
index 42427814c..0e42defa5 100644
--- a/test/srt/test_vision_openai_server_b.py
+++ b/test/srt/test_vision_openai_server_b.py
@@ -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"