wsxiaoys commited on
Commit
c6065b8
·
1 Parent(s): b50d0b0

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -1,8 +1,8 @@
1
- FROM tabbyml/tabby:0.1.2
2
 
3
  USER root
4
  RUN mkdir -p /data
5
  RUN chown 1000 /data
6
 
7
  USER 1000
8
- CMD ["serve", "--device", "cuda", "--model", "TabbyML/SantaCoder-1B"]
 
1
+ FROM tabbyml/tabby:0.2.0
2
 
3
  USER root
4
  RUN mkdir -p /data
5
  RUN chown 1000 /data
6
 
7
  USER 1000
8
+ CMD ["serve", "--device", "cuda", "--model", "TabbyML/StarCoder-1B", "--chat-model", "TabbyML/Mistral-7B"]