Update docker_base for meta-reference-gpu

This commit is contained in:
Ashwin Bharambe 2024-10-25 09:13:33 -07:00 committed by Xi Yan
parent be3adb0964
commit 392c23e557

View file

@ -0,0 +1,14 @@
name: meta-reference-gpu
distribution_spec:
docker_image: pytorch/pytorch:2.5.0-cuda12.4-cudnn9-runtime
description: Use code from `llama_stack` itself to serve all llama stack APIs
providers:
inference: meta-reference
memory:
- meta-reference
- remote::chromadb
- remote::pgvector
safety: meta-reference
agents: meta-reference
telemetry: meta-reference
image_type: docker