Adding verl support

Author: Harsh Thakkar

src/sagemaker/modules/train/model_trainer.py

@@ -649,7 +649,7 @@ def _create_training_job_args(
                 key_prefix=input_data_key_prefix,
             )
             final_input_data_config.append(recipe_channel)
-            if self._is_nova_recipe:
+            if self._is_nova_recipe or self._is_llmft_recipe:
                 self.hyperparameters.update(
                     {"sagemaker_recipe_local_path": SM_RECIPE_CONTAINER_PATH}
                 )

src/sagemaker/modules/train/sm_recipes/utils.py

@@ -289,7 +289,7 @@ def _is_llmft_recipe(
 
     A recipe is considered a LLMFT recipe if it meets the following conditions:
         1. Having a run section
-        2. The model_type in run is llmft
+        2. The model_type in run is llm_finetuning_aws or verl
         3. Having a training_config section
 
     Args:
@@ -299,8 +299,10 @@ def _is_llmft_recipe(
         bool: True if the recipe is a LLMFT recipe, False otherwise
     """
     run_config = recipe.get("run", {})
-    has_llmft_model = run_config.get("model_type", "").lower() == "llm_finetuning_aws"
-    return bool(has_llmft_model) and bool(recipe.get("training_config"))
+    model_type = run_config.get("model_type", "").lower()
+    has_llmft_model = model_type == "llm_finetuning_aws"
+    has_verl_model = model_type == "verl"
+    return (bool(has_llmft_model) or bool(has_verl_model)) and bool(recipe.get("training_config"))
 
 
 def _get_args_from_nova_recipe(

tests/unit/sagemaker/modules/train/sm_recipes/test_utils.py

@@ -545,12 +545,34 @@ def test_get_args_from_nova_recipe_with_evaluation(test_case):
             },
             "is_llmft": False,
         },
+        {
+            "recipe": {
+                "run": {
+                    "name": "verl-grpo-llama",
+                    "model_type": "verl",
+                },
+                "trainer": {"num_nodes": "1"},
+                "training_config": {"trainer": {"total_epochs": 2}},
+            },
+            "is_llmft": True,
+        },
+        {
+            "recipe": {
+                "run": {
+                    "name": "verl-grpo-llama",
+                    "model_type": "verl",
+                },
+            },
+            "is_llmft": False,
+        },
     ],
     ids=[
         "llmft_model",
         "llmft_model_subtype",
         "llmft_missing_training_config",
         "non_llmft_model",
+        "verl_model",
+        "verl_missing_training_config",
     ],
 )
 def test_is_llmft_recipe(test_case):