diff --git a/src/pages/llmodels/components/deploy-builtin-modal.tsx b/src/pages/llmodels/components/deploy-builtin-modal.tsx index 892ce5ff..f1ee1842 100644 --- a/src/pages/llmodels/components/deploy-builtin-modal.tsx +++ b/src/pages/llmodels/components/deploy-builtin-modal.tsx @@ -96,6 +96,7 @@ const AddModal: React.FC = (props) => { const selectSpecRef = useRef({} as CatalogSpec); const specListRef = useRef([]); const noCompatibleGPUsRef = useRef(false); + const [noCompatibleGPUs, setNoCompatibleGPUs] = useState(false); const handleSumit = () => { form.current?.submit?.(); @@ -265,6 +266,7 @@ const AddModal: React.FC = (props) => { // If no avaliable gpus for the model, show warning message if (!res.items.length) { noCompatibleGPUsRef.current = true; + setNoCompatibleGPUs(true); setWarningStatus({ show: true, type: 'warning', @@ -273,6 +275,7 @@ const AddModal: React.FC = (props) => { return; } noCompatibleGPUsRef.current = false; + setNoCompatibleGPUs(false); handleCheckCompatibility(allValues); } catch (error) { // ignore @@ -391,7 +394,11 @@ const AddModal: React.FC = (props) => { showOkBtn={!showExtraButton} extra={ showExtraButton && ( -