Hugging Face's logo

Spaces:
wsxiaoys
/
hf-tabby
Runtime error

App Files Community
Meng Zhang commited on
Commit
3f65b2e
•
1 Parent(s): 7198afe

enable cuda support

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -1,3 +1,3 @@
1
  FROM tabbyml/tabby:latest
2
 
3
- CMD ["serve", "--model", "TabbyML/SantaCoder-1B"]
 
1
  FROM tabbyml/tabby:latest
2
 
3
+ CMD ["serve", "--model", "TabbyML/SantaCoder-1B", "--device", "cuda"]