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

Merge branch 'main' into enable-vector-stores-files-api-tests

Browse source
This commit is contained in:
Francisco Arceo 2025-07-31 14:28:11 -04:00 • committed by GitHub
commit 213bd1b1f7
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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 @pytest.mark.timeout(360) # 6 minutes timeout
@skip_because_resource_intensive
def test_supervised_fine_tune(self, llama_stack_client, purpose, source): def test_supervised_fine_tune(self, llama_stack_client, purpose, source):
logger.info("Starting supervised fine-tuning test") logger.info("Starting supervised fine-tuning test")
@ -111,7 +110,7 @@ class TestPostTraining:
# train with HF trl SFTTrainer as the default # train with HF trl SFTTrainer as the default
_ = llama_stack_client.post_training.supervised_fine_tune( _ = llama_stack_client.post_training.supervised_fine_tune(
job_uuid=job_uuid, job_uuid=job_uuid,
model="ibm-granite/granite-3.3-2b-instruct", model="HuggingFaceTB/SmolLM2-135M-Instruct", # smaller model that supports the current sft recipe
algorithm_config=algorithm_config, algorithm_config=algorithm_config,
training_config=training_config, training_config=training_config,
hyperparam_search_config={}, hyperparam_search_config={},