diff --git a/ui/src/workflow/nodes/ai-chat-node/index.vue b/ui/src/workflow/nodes/ai-chat-node/index.vue
index 32347b2d07e..51d75705625 100644
--- a/ui/src/workflow/nodes/ai-chat-node/index.vue
+++ b/ui/src/workflow/nodes/ai-chat-node/index.vue
@@ -44,19 +44,17 @@
             </div>
           </template>
           <div class="flex-between w-full" v-if="chat_data.model_id_type !== 'reference'">
-            <div>
-              <ModelSelect
-                @change="model_change"
-                @wheel="wheel"
-                :teleported="false"
-                v-model="chat_data.model_id"
-                :placeholder="$t('views.application.form.aiModel.placeholder')"
-                :options="modelOptions"
-                @submitModel="getSelectModel"
-                showFooter
-                :model-type="'LLM'"
-              ></ModelSelect>
-            </div>
+            <ModelSelect
+              @change="model_change"
+              @wheel="wheel"
+              :teleported="false"
+              v-model="chat_data.model_id"
+              :placeholder="$t('views.application.form.aiModel.placeholder')"
+              :options="modelOptions"
+              @submitModel="getSelectModel"
+              showFooter
+              :model-type="'LLM'"
+            ></ModelSelect>
             <div class="ml-8">
               <el-button
                 :disabled="!chat_data.model_id"