Dockerfile

FROM ghcr.io/huggingface/text-generation-inference:latest

# Small modell sshleifer/tiny-gpt2
CMD ["--model-id", "sshleifer/tiny-gpt2"]