diff --git a/website/docs/installation/docker-compose.mdx b/website/docs/installation/docker-compose.mdx
index 9f79d67..8dab5c4 100644
--- a/website/docs/installation/docker-compose.mdx
+++ b/website/docs/installation/docker-compose.mdx
@@ -9,27 +9,11 @@ import Tabs from '@theme/Tabs';
import TabItem from '@theme/TabItem';
-
-
-```yaml title="docker-compose.yml"
-version: '3.5'
-
-services:
- tabby:
- restart: always
- image: tabbyml/tabby
- command: serve --model TabbyML/StarCoder-1B
- volumes:
- - "$HOME/.tabby:/data"
- ports:
- - 8080:8080
-```
-
-
```yaml title="docker-compose.yml"
version: '3.5'
+
services:
tabby:
restart: always
@@ -48,6 +32,23 @@ services:
capabilities: [gpu]
```
+
+
+
+```yaml title="docker-compose.yml"
+version: '3.5'
+
+services:
+ tabby:
+ restart: always
+ image: tabbyml/tabby
+ command: serve --model TabbyML/StarCoder-1B
+ volumes:
+ - "$HOME/.tabby:/data"
+ ports:
+ - 8080:8080
+```
+
diff --git a/website/docs/installation/docker.mdx b/website/docs/installation/docker.mdx
index 31c78b0..7f26f87 100644
--- a/website/docs/installation/docker.mdx
+++ b/website/docs/installation/docker.mdx
@@ -10,18 +10,18 @@ import Tabs from '@theme/Tabs';
import TabItem from '@theme/TabItem';
-
-
- ```bash title="run.sh"
- docker run -it -p 8080:8080 -v $HOME/.tabby:/data tabbyml/tabby serve --model TabbyML/StarCoder-1B
- ```
-
-
```bash title="run.sh"
docker run -it --gpus all -p 8080:8080 -v $HOME/.tabby:/data tabbyml/tabby serve --model TabbyML/StarCoder-1B --device cuda
```
+
+
+
+ ```bash title="run.sh"
+ docker run -it -p 8080:8080 -v $HOME/.tabby:/data tabbyml/tabby serve --model TabbyML/StarCoder-1B
+ ```
+