Fix checkpoint path

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

Author: chtruong814

tests/functional_tests/tests_inframework/test_deploy_query_mbridge_ray.py

@@ -56,7 +56,7 @@ def test_deploy_ray(self):
                     "--parallel-mode",
                     "scripts/deploy/nlp/deploy_ray_inframework.py",
                     "--megatron_checkpoint",
-                    mbridge_checkpoint_path,
+                    f"{mbridge_checkpoint_path}/iter_0000000",
                     "--model_id",
                     "llama",
                     "--num_gpus",

tests/functional_tests/tests_inframework/test_deploy_query_vlm_ray.py

@@ -80,7 +80,7 @@ def test_deploy_ray(self):
                     "--parallel-mode",
                     "scripts/deploy/multimodal/deploy_ray_inframework.py",
                     "--megatron_checkpoint",
-                    vlm_checkpoint_path,
+                    f"{vlm_checkpoint_path}/iter_0000000",
                     "--model_id",
                     "megatron-multimodal",
                     "--num_gpus",