fix(tgi): add label

huggingface · Sep 24, 2024 · 38904a3 · 38904a3
1 parent 9313245
commit 38904a3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/text-generation-inference/Dockerfile b/text-generation-inference/Dockerfile
@@ -142,6 +142,7 @@ ENTRYPOINT ["./entrypoint.sh"]
 
 # Final image
 FROM neuron
+LABEL org.opencontainers.image.source=https://github.com/huggingface/optimum-neuron
 COPY text-generation-inference/tgi-entrypoint.sh text-generation-inference/tgi_env.py /
 ENTRYPOINT ["/tgi-entrypoint.sh"]
 CMD ["--json-output"]