From f5dcc03742d2bc0cba90a0874630e58e948acfef Mon Sep 17 00:00:00 2001
From: Xi Yan <xiyan@meta.com>
Date: Wed, 23 Oct 2024 20:22:00 -0700
Subject: [PATCH] use pytorch/pytorch as base

---
 distributions/meta-reference-gpu/build.yaml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/distributions/meta-reference-gpu/build.yaml b/distributions/meta-reference-gpu/build.yaml
index e76197330..5b1521a92 100644
--- a/distributions/meta-reference-gpu/build.yaml
+++ b/distributions/meta-reference-gpu/build.yaml
@@ -1,5 +1,6 @@
 name: meta-reference-gpu
 distribution_spec:
+  docker_image: pytorch/pytorch
   description: Use code from `llama_stack` itself to serve all llama stack APIs
   providers:
     inference: meta-reference