phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-15 14:08:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

changed to smaller model

Browse source
This commit is contained in:
Nehanth 2025-07-31 16:46:31 +00:00
parent ffb6306fbd
commit 48110f2d0c
1 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
tests/integration/post_training/test_post_training.py
View file

@ -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={},