From 48110f2d0c1dbde05a2aa99a2cc685c7f6ab17b1 Mon Sep 17 00:00:00 2001 From: Nehanth Date: Thu, 31 Jul 2025 16:46:31 +0000 Subject: [PATCH] changed to smaller model --- tests/integration/post_training/test_post_training.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/integration/post_training/test_post_training.py b/tests/integration/post_training/test_post_training.py index 002da1160..1db2a8940 100644 --- a/tests/integration/post_training/test_post_training.py +++ b/tests/integration/post_training/test_post_training.py @@ -70,7 +70,6 @@ class TestPostTraining: ], ) @pytest.mark.timeout(360) # 6 minutes timeout - @skip_because_resource_intensive def test_supervised_fine_tune(self, llama_stack_client, purpose, source): logger.info("Starting supervised fine-tuning test") @@ -111,7 +110,7 @@ class TestPostTraining: # train with HF trl SFTTrainer as the default _ = llama_stack_client.post_training.supervised_fine_tune( job_uuid=job_uuid, - model="ibm-granite/granite-3.3-2b-instruct", + model="HuggingFaceTB/SmolLM2-135M-Instruct", # added a smaller model that supports the current sft recipe algorithm_config=algorithm_config, training_config=training_config, hyperparam_search_config={},