Update docker-compose.mdx

TabbyML · Dec 13, 2023 · 2c55ee4 · 2c55ee4
1 parent 6fc195e
commit 2c55ee4
Showing 1 changed file with 2 additions and 21 deletions.
diff --git a/website/docs/installation/docker-compose.mdx b/website/docs/installation/docker-compose.mdx
@@ -16,8 +16,8 @@ version: '3.5'
 
 services:
   tabby:
-    restart: unless-stopped
-    image: tabbyml/tabby-cuda
+    restart: always
+    image: tabbyml/tabby
     command: serve --model TabbyML/StarCoder-1B --device cuda
     volumes:
       - "$HOME/.tabby:/data"
@@ -33,25 +33,6 @@ services:
 ```
 
   </TabItem>
-  <TabItem value="rocm" label="ROCm">
-
-```yaml title="docker-compose.yml"
-version: '3.5'
-services:
-  tabby:
-    restart: unless-stopped
-    image: tabbyml/tabby-rocm
-    command: serve --model TabbyML/StarCoder-1B --device rocm
-    volumes:
-      - "$HOME/.tabby:/data"
-    ports:
-      - 8080:8080
-    devices:
-      - /dev/dri
-      - /dev/kfd
-```
-
-    </TabItem>
   <TabItem value="cpu" label="CPU">
 
 ```yaml title="docker-compose.yml"