From a087677c6e27aaea70a0439bc8139ce3462b7c6e Mon Sep 17 00:00:00 2001
From: jialin <jialinkuang@126.com>
Date: Tue, 8 Apr 2025 09:46:37 +0800
Subject: [PATCH] chore: allow to  config distribution in vllm

---
 .../llmodels/components/advance-config.tsx    | 41 ++++++++++---------
 1 file changed, 22 insertions(+), 19 deletions(-)

diff --git a/src/pages/llmodels/components/advance-config.tsx b/src/pages/llmodels/components/advance-config.tsx
index 8c09e03c..9e469b27 100644
--- a/src/pages/llmodels/components/advance-config.tsx
+++ b/src/pages/llmodels/components/advance-config.tsx
@@ -453,25 +453,28 @@ const AdvanceConfig: React.FC<AdvanceConfigProps> = (props) => {
             </Form.Item>
           </div>
         )}
-        {scheduleType === 'auto' && backend === backendOptionsMap.llamaBox && (
-          <div style={{ paddingBottom: 22, paddingLeft: 10 }}>
-            <Form.Item<FormData>
-              name="distributed_inference_across_workers"
-              valuePropName="checked"
-              style={{ padding: '0 10px', marginBottom: 0 }}
-              noStyle
-            >
-              <CheckboxField
-                title={intl.formatMessage({
-                  id: 'models.form.distribution.tips'
-                })}
-                label={intl.formatMessage({
-                  id: 'resources.form.enableDistributedInferenceAcrossWorkers'
-                })}
-              ></CheckboxField>
-            </Form.Item>
-          </div>
-        )}
+        {scheduleType === 'auto' &&
+          [backendOptionsMap.llamaBox, backendOptionsMap.vllm].includes(
+            backend
+          ) && (
+            <div style={{ paddingBottom: 22, paddingLeft: 10 }}>
+              <Form.Item<FormData>
+                name="distributed_inference_across_workers"
+                valuePropName="checked"
+                style={{ padding: '0 10px', marginBottom: 0 }}
+                noStyle
+              >
+                <CheckboxField
+                  title={intl.formatMessage({
+                    id: 'models.form.distribution.tips'
+                  })}
+                  label={intl.formatMessage({
+                    id: 'resources.form.enableDistributedInferenceAcrossWorkers'
+                  })}
+                ></CheckboxField>
+              </Form.Item>
+            </div>
+          )}
         <div style={{ paddingBottom: 22, paddingLeft: 10 }}>
           <Form.Item<FormData>
             name="restart_on_error"