Skip to content

Commit bdae772

Browse files
committed
Fix vlm test
Signed-off-by: Charlie Truong <chtruong@nvidia.com>
1 parent e25cdfc commit bdae772

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

‎tests/functional_tests/tests_inframework/test_deploy_query_vlm_ray.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ def test_deploy_ray(self):
6868
from megatron.bridge import AutoBridge
6969

7070
vlm_checkpoint_path = "/opt/checkpoints/hf_qwen25_vl_3b_mbridge"
71-
AutoBridge.import_ckpt("Qwen/Qwen2.5-VL-3B-Instruct", mbridge_checkpoint_path)
71+
AutoBridge.import_ckpt("Qwen/Qwen2.5-VL-3B-Instruct", vlm_checkpoint_path)
7272

7373
# Run Ray deployment for Megatron multimodal (VLM) model
7474
self.deploy_proc = subprocess.Popen(

0 commit comments

Comments
 (0)