This commit is contained in:
Xi Yan 2024-10-22 20:00:43 -07:00
parent 821810657f
commit aefa84e70a
5 changed files with 117 additions and 54 deletions

View file

@ -36,7 +36,7 @@ def available_providers() -> List[ProviderSpec]:
pip_packages=(
META_REFERENCE_DEPS
+ [
"fbgemm-gpu==0.8.0",
"fbgemm-gpu==1.0.0",
]
),
module="llama_stack.providers.impls.meta_reference.inference",