From 1d5f7acf966976e211497e5c580d6ff17faafeb1 Mon Sep 17 00:00:00 2001 From: Jennifer Player Date: Tue, 27 Feb 2024 09:48:41 -0500 Subject: [PATCH] updated to use new import model mutation --- .../AddModelPanel/ScanModels/ScanModelsResults.tsx | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/ScanModels/ScanModelsResults.tsx b/invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/ScanModels/ScanModelsResults.tsx index b8c6efd938..960e00c2a6 100644 --- a/invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/ScanModels/ScanModelsResults.tsx +++ b/invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/ScanModels/ScanModelsResults.tsx @@ -16,7 +16,7 @@ import type { ChangeEventHandler } from 'react'; import { useCallback, useMemo, useState } from 'react'; import { useTranslation } from 'react-i18next'; import { PiXBold } from 'react-icons/pi'; -import { type ScanFolderResponse, useImportMainModelsMutation } from 'services/api/endpoints/models'; +import { type ScanFolderResponse, useInstallModelMutation } from 'services/api/endpoints/models'; import { ScanModelResultItem } from './ScanModelResultItem'; @@ -29,7 +29,7 @@ export const ScanModelsResults = ({ results }: ScanModelResultsProps) => { const [searchTerm, setSearchTerm] = useState(''); const dispatch = useAppDispatch(); - const [importMainModel] = useImportMainModelsMutation(); + const [installModel] = useInstallModelMutation(); const filteredResults = useMemo(() => { return results.filter((result) => { @@ -51,7 +51,7 @@ export const ScanModelsResults = ({ results }: ScanModelResultsProps) => { if (result.is_installed) { continue; } - importMainModel({ source: result.path, config: undefined }) + installModel({ source: result.path }) .unwrap() .then((_) => { dispatch( @@ -76,7 +76,7 @@ export const ScanModelsResults = ({ results }: ScanModelResultsProps) => { } }); } - }, [importMainModel, filteredResults, dispatch, t]); + }, [installModel, filteredResults, dispatch, t]); return ( <>