diff --git a/Dockerfile b/Dockerfile
index eb2ec68..7bc6dcc 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -48,11 +48,12 @@ COPY --from=s3build /app/dist /dist
 #CMD ["node", "dist/start/server.js"]
 
 #============================================ chatai-ui ========================================================
-FROM node:18.20.6 AS chataibuilder
+FROM nvcr.io/nvidia/tritonserver:24.04-py3-min AS chataibuilder
 
-# 安装指定版本的 npm 和 pm2
-RUN npm install -g npm@10.8.2 \
-    && npm install -g pm2@5.4.3
+RUN curl -fsSL https://deb.nodesource.com/setup_18.x | bash - && \
+    apt-get update && \
+    apt-get install -y nodejs && \
+    npm install -g npm@10.8.2 pnpm@10.9.0
 
 WORKDIR /app