From c30a107b82b6df05076e73e86ebb0104f3fb3c51 Mon Sep 17 00:00:00 2001 From: Botao Chen Date: Thu, 16 Jan 2025 14:50:28 -0800 Subject: [PATCH] commit --- 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 3c5d06c05..0efe623c9 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", "torchao", "numpy"], + pip_packages=["torch", "torchtune==0.4.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=[