Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

updated to use new import model mutation

Browse Source
This commit is contained in:
Jennifer Player 2024-02-27 09:48:41 -05:00
parent 175cfe41a4
commit 110b0bc8fe
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/ScanModels/ScanModelsResults.tsx
View File

@ -16,7 +16,7 @@ import type { ChangeEventHandler } from 'react';
import { useCallback, useMemo, useState } from 'react'; import { useCallback, useMemo, useState } from 'react';
import { useTranslation } from 'react-i18next'; import { useTranslation } from 'react-i18next';
import { PiXBold } from 'react-icons/pi'; 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'; import { ScanModelResultItem } from './ScanModelResultItem';
@ -29,7 +29,7 @@ export const ScanModelsResults = ({ results }: ScanModelResultsProps) => {
const [searchTerm, setSearchTerm] = useState(''); const [searchTerm, setSearchTerm] = useState('');
const dispatch = useAppDispatch(); const dispatch = useAppDispatch();
const [importMainModel] = useImportMainModelsMutation(); const [installModel] = useInstallModelMutation();
const filteredResults = useMemo(() => { const filteredResults = useMemo(() => {
return results.filter((result) => { return results.filter((result) => {
@ -51,7 +51,7 @@ export const ScanModelsResults = ({ results }: ScanModelResultsProps) => {
if (result.is_installed) { if (result.is_installed) {
continue; continue;
} }
importMainModel({ source: result.path, config: undefined }) installModel({ source: result.path })
.unwrap() .unwrap()
.then((_) => { .then((_) => {
dispatch( dispatch(
@ -76,7 +76,7 @@ export const ScanModelsResults = ({ results }: ScanModelResultsProps) => {
} }
}); });
} }
}, [importMainModel, filteredResults, dispatch, t]); }, [installModel, filteredResults, dispatch, t]);
return ( return (
<> <>