Use pre-created checkpoint

Signed-off-by: Charlie Truong <chtruong@nvidia.com>

Author: chtruong814

tests/functional_tests/tests_inframework/test_deploy_query_mbridge_ray.py

@@ -40,11 +40,7 @@ def teardown_method(self):
 
     def test_deploy_ray(self):
         try:
-            # Export Megatron-Bridge checkpoint
-            from megatron.bridge import AutoBridge
-
-            mbridge_checkpoint_path = "/opt/checkpoints/hf_llama32_1B_mbridge"
-            AutoBridge.import_ckpt("meta-llama/Llama-3.2-1B", mbridge_checkpoint_path)
+            mbridge_checkpoint_path = "/home/TestData/export-deploy/checkpoints/mbridge/hf_llama32_1B_mbridge/iter_0000000"
 
             # Run Ray deployment
             self.deploy_proc = subprocess.Popen(
@@ -56,7 +52,7 @@ def test_deploy_ray(self):
                     "--parallel-mode",
                     "scripts/deploy/nlp/deploy_ray_inframework.py",
                     "--megatron_checkpoint",
-                    f"{mbridge_checkpoint_path}/iter_0000000",
+                    mbridge_checkpoint_path,
                     "--model_id",
                     "llama",
                     "--num_gpus",

tests/functional_tests/tests_inframework/test_deploy_query_vlm_ray.py

@@ -64,11 +64,7 @@ def teardown_method(self):
 
     def test_deploy_ray(self):
         try:
-            # Export Megatron-Bridge checkpoint
-            from megatron.bridge import AutoBridge
-
-            vlm_checkpoint_path = "/opt/checkpoints/hf_qwen25_vl_3b_mbridge"
-            AutoBridge.import_ckpt("Qwen/Qwen2.5-VL-3B-Instruct", vlm_checkpoint_path)
+            vlm_checkpoint_path = "/home/TestData/export-deploy/checkpoints/mbridge/hf_qwen25_vl_3b_mbridge/iter_0000000"
 
             # Run Ray deployment for Megatron multimodal (VLM) model
             self.deploy_proc = subprocess.Popen(
@@ -80,7 +76,7 @@ def test_deploy_ray(self):
                     "--parallel-mode",
                     "scripts/deploy/multimodal/deploy_ray_inframework.py",
                     "--megatron_checkpoint",
-                    f"{vlm_checkpoint_path}/iter_0000000",
+                    vlm_checkpoint_path,
                     "--model_id",
                     "megatron-multimodal",
                     "--num_gpus",