phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2026-01-03 08:12:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

debugged impl errors for building container and running data prep

Browse source 
Signed-off-by: James Kunstle <jkunstle@redhat.com>
This commit is contained in:
James Kunstle 2025-03-14 00:42:20 -07:00
parent 06465441f2
commit 68000499f7
6 changed files with 66 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/providers/registry/post_training.py
View file

@ -14,7 +14,7 @@ def available_providers() -> List[ProviderSpec]:
InlineProviderSpec(
api=Api.post_training,
provider_type="inline::torchtune",
pip_packages=["torch", "torchtune==0.5.0", "torchao==0.8.0", "numpy"],
pip_packages=["torch", "torchtune==0.5.0", "torchao==0.8.0", "numpy", "openai"],
module="llama_stack.providers.inline.post_training.torchtune",
config_class="llama_stack.providers.inline.post_training.torchtune.TorchtunePostTrainingConfig",
api_dependencies=[
@ -25,7 +25,7 @@ def available_providers() -> List[ProviderSpec]:
InlineProviderSpec(
api=Api.post_training,
provider_type="inline::huggingface-ilab",
pip_packages=["torch", "transformers", "datasets", "numpy"],
pip_packages=["torch", "transformers", "datasets", "numpy", "openai"],
module="llama_stack.providers.inline.post_training.huggingface_ilab",
config_class="llama_stack.providers.inline.post_training.huggingface_ilab.HFilabPostTrainingConfig",
api_dependencies=[