From c84eecd1673c235b1fba66272c74f525a0795fac Mon Sep 17 00:00:00 2001 From: Botao Chen Date: Thu, 16 Jan 2025 15:08:05 -0800 Subject: [PATCH] update --- llama_stack/providers/registry/post_training.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llama_stack/providers/registry/post_training.py b/llama_stack/providers/registry/post_training.py index 0efe623c9..3bcda6508 100644 --- a/llama_stack/providers/registry/post_training.py +++ b/llama_stack/providers/registry/post_training.py @@ -14,7 +14,7 @@ def available_providers() -> List[ProviderSpec]: InlineProviderSpec( api=Api.post_training, provider_type="inline::torchtune", - pip_packages=["torch", "torchtune==0.4.0", "torchao==0.8.0", "numpy"], + pip_packages=["torch", "torchtune==0.5.0", "torchao==0.8.0", "numpy"], module="llama_stack.providers.inline.post_training.torchtune", config_class="llama_stack.providers.inline.post_training.torchtune.TorchtunePostTrainingConfig", api_dependencies=[