diff --git a/installer/Dockerfile b/installer/Dockerfile
index 9149cf7b1..96c89b64b 100644
--- a/installer/Dockerfile
+++ b/installer/Dockerfile
@@ -25,7 +25,7 @@ RUN python3 -m venv /opt/py3 && \
     pip install poetry --break-system-packages && \
     poetry config virtualenvs.create false && \
     . /opt/py3/bin/activate && \
-    if [ "$(uname -m)" != "x86_64" ]; then sed -i '/^torch/d' pyproject.toml; fi && \
+    if [ "$(uname -m)" = "x86_64" ]; then sed -i 's/^torch.*/torch = {version = "^2.2.1+cpu", source = "pytorch"}/g' pyproject.toml; fi && \
     poetry install
 
 FROM ghcr.io/1panel-dev/maxkb-python-pg:python3.11-pg15.6
diff --git a/pyproject.toml b/pyproject.toml
index bee1b9cf5..2675941b2 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -18,7 +18,7 @@ diskcache = "^5.6.3"
 pillow = "^10.2.0"
 filetype = "^1.2.0"
 chardet = "^5.2.0"
-torch = {version = "^2.2.1+cpu", source = "pytorch"}
+torch = "^2.2.1"
 sentence-transformers = "^2.2.2"
 blinker = "^1.6.3"
 openai = "^1.13.3"