sshao45 commited on
Commit
c3eb807
·
1 Parent(s): 9bc70a0
Files changed (1) hide show
  1. Dockerfile +4 -0
Dockerfile ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ FROM ghcr.io/huggingface/text-generation-inference:latest
2
+
3
+ # Small modell sshleifer/tiny-gpt2
4
+ CMD ["--model-id", "sshleifer/tiny-gpt2"]