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"]