From 4f821f0f8bbe3ef1fc6466cc74bd5092294622c6 Mon Sep 17 00:00:00 2001
From: Yannic Kilcher <yk@users.noreply.github.com>
Date: Thu, 9 Feb 2023 00:11:05 +0100
Subject: [PATCH] switched to HF text-generation-inference

---
 docker-compose.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docker-compose.yaml b/docker-compose.yaml
index 5d9955bf..3cfb4c21 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -189,7 +189,7 @@ services:
     profiles: ["inference"]
 
   inference-text-generation-server:
-    image: ykilcher/text-generation-inference
+    image: ghcr.io/huggingface/text-generation-inference
     environment:
       - "MODEL_ID=distilgpt2"
     profiles: ["inference"]