GKV_TGI_Inference / Dockerfile
sshao45's picture
docker
c3eb807
raw
history blame contribute delete
135 Bytes
FROM ghcr.io/huggingface/text-generation-inference:latest
# Small modell sshleifer/tiny-gpt2
CMD ["--model-id", "sshleifer/tiny-gpt2"]