Update docker_base for meta-reference-gpu

This commit is contained in:
Ashwin Bharambe 2024-10-25 09:13:33 -07:00
parent 70d59b0f5d
commit a387ca22e2

View file

@ -1,6 +1,6 @@
name: meta-reference-gpu
distribution_spec:
docker_image: pytorch/pytorch
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