phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-05 04:17:32 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Remove additional_pip_packages; move deps to providers

Browse source
This commit is contained in:
Ashwin Bharambe 2024-08-08 10:19:46 -07:00
parent 6de36b6a15
commit 8d7ecf0c47
5 changed files with 18 additions and 32 deletions
Download patch file Download diff file Expand all files Collapse all files

6
llama_toolchain/inference/providers.py
View file

@ -15,7 +15,13 @@ def available_inference_providers() -> List[ProviderSpec]:
api=Api.inference,
provider_id="meta-reference",
pip_packages=[
"accelerate",
"blobfile",
"codeshield",
"fairscale",
"fbgemm-gpu==0.8.0",
"torch",
"transformers",
"zmq",
],
module="llama_toolchain.inference.meta_reference",