address comments

2025-12-17 17:32:42 +00:00 · 2024-12-10 20:50:17 -08:00 · 2024-12-10 20:50:17 -08:00 · 68ebf8a8da
commit 68ebf8a8da
parent 9c1ae088f9
6 changed files with 97 additions and 29 deletions
--- a/llama_stack/providers/registry/post_training.py
+++ b/llama_stack/providers/registry/post_training.py
@ -19,6 +19,7 @@ def available_providers() -> List[ProviderSpec]:
            config_class="llama_stack.providers.inline.post_training.torchtune.TorchtunePostTrainingConfig",
            api_dependencies=[
                Api.datasetio,
+                Api.datasets,
            ],
        ),
    ]