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 add model copy, added search to hugging face results

Browse Source
This commit is contained in:
Jennifer Player 2024-03-11 16:30:32 -04:00 committed by psychedelicious
parent 90340a39c7
commit 2a300ecada
3 changed files with 14 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/frontend/web/public/locales/en.json
View File

@ -1409,7 +1409,7 @@
"maskSentControlnetAssets": "Mask Sent to ControlNet & Assets", "maskSentControlnetAssets": "Mask Sent to ControlNet & Assets",
"metadataLoadFailed": "Failed to load metadata", "metadataLoadFailed": "Failed to load metadata",
"modelAdded": "Model Added: {{modelName}}", "modelAdded": "Model Added: {{modelName}}",
"modelAddedSimple": "Model Added", "modelAddedSimple": "Model Added to Queue",
"modelAddFailed": "Model Add Failed", "modelAddFailed": "Model Add Failed",
"modelImportCanceled": "Model Import Canceled", "modelImportCanceled": "Model Import Canceled",
"modelImportRemoved": "Model Import Removed", "modelImportRemoved": "Model Import Removed",

17
invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/HuggingFaceFolder/HuggingFaceResults.tsx
View File

@ -13,7 +13,7 @@ import ScrollableContent from 'common/components/OverlayScrollbars/ScrollableCon
import { addToast } from 'features/system/store/systemSlice'; import { addToast } from 'features/system/store/systemSlice';
import { makeToast } from 'features/system/util/makeToast'; import { makeToast } from 'features/system/util/makeToast';
import type { ChangeEventHandler } from 'react'; import type { ChangeEventHandler } from 'react';
import { useCallback, 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 { useInstallModelMutation } from 'services/api/endpoints/models'; import { useInstallModelMutation } from 'services/api/endpoints/models';
@ -31,6 +31,13 @@ export const HuggingFaceResults = ({ results }: HuggingFaceResultsProps) => {
const [installModel] = useInstallModelMutation(); const [installModel] = useInstallModelMutation();
const filteredResults = useMemo(() => {
return results.filter((result) => {
const modelName = result.split('/').slice(-1)[0];
return modelName?.toLowerCase().includes(searchTerm.toLowerCase());
});
}, [results, searchTerm]);
const handleSearch: ChangeEventHandler<HTMLInputElement> = useCallback((e) => { const handleSearch: ChangeEventHandler<HTMLInputElement> = useCallback((e) => {
setSearchTerm(e.target.value.trim()); setSearchTerm(e.target.value.trim());
}, []); }, []);
@ -40,7 +47,7 @@ export const HuggingFaceResults = ({ results }: HuggingFaceResultsProps) => {
}, []); }, []);
const handleAddAll = useCallback(() => { const handleAddAll = useCallback(() => {
for (const result of results) { for (const result of filteredResults) {
installModel({ source: result }) installModel({ source: result })
.unwrap() .unwrap()
.then((_) => { .then((_) => {
@ -66,7 +73,7 @@ export const HuggingFaceResults = ({ results }: HuggingFaceResultsProps) => {
} }
}); });
} }
}, [results, installModel, dispatch, t]); }, [filteredResults, installModel, dispatch, t]);
return ( return (
<> <>
@ -77,7 +84,7 @@ export const HuggingFaceResults = ({ results }: HuggingFaceResultsProps) => {
{t('modelManager.availableModels')} {t('modelManager.availableModels')}
</Heading> </Heading>
<Flex alignItems="center" gap="4"> <Flex alignItems="center" gap="4">
<Button onClick={handleAddAll} isDisabled={results.length === 0}> <Button size="sm" onClick={handleAddAll} isDisabled={results.length === 0}>
{t('modelManager.addAll')} {t('modelManager.addAll')}
</Button> </Button>
<InputGroup maxW="300px" size="xs"> <InputGroup maxW="300px" size="xs">
@ -106,7 +113,7 @@ export const HuggingFaceResults = ({ results }: HuggingFaceResultsProps) => {
<Flex height="100%" layerStyle="third" borderRadius="base" p={4} mt={4} mb={4}> <Flex height="100%" layerStyle="third" borderRadius="base" p={4} mt={4} mb={4}>
<ScrollableContent> <ScrollableContent>
<Flex flexDir="column" gap={3}> <Flex flexDir="column" gap={3}>
{results.map((result) => ( {filteredResults.map((result) => (
<HuggingFaceResultItem key={result} result={result} /> <HuggingFaceResultItem key={result} result={result} />
))} ))}
</Flex> </Flex>

2
invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/ScanFolder/ScanFolderResults.tsx
View File

@ -87,7 +87,7 @@ export const ScanModelsResults = ({ results }: ScanModelResultsProps) => {
{t('modelManager.scanResults')} {t('modelManager.scanResults')}
</Heading> </Heading>
<Flex alignItems="center" gap="4"> <Flex alignItems="center" gap="4">
<Button onClick={handleAddAll} isDisabled={filteredResults.length === 0}> <Button size="sm" onClick={handleAddAll} isDisabled={filteredResults.length === 0}>
{t('modelManager.addAll')} {t('modelManager.addAll')}
</Button> </Button>
<InputGroup maxW="300px" size="xs"> <InputGroup maxW="300px" size="xs">