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

feat(ui): improve starter model search for spandrel models

Browse Source
This commit is contained in:
psychedelicious 2024-07-24 07:31:36 +10:00
parent b4519ea61f
commit 94b5b2a467
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/StarterModels/StarterModelsResults.tsx
View File

@ -19,9 +19,16 @@ export const StarterModelsResults = ({ results }: StarterModelsResultsProps) =>
const filteredResults = useMemo(() => { const filteredResults = useMemo(() => {
return results.filter((result) => { return results.filter((result) => {
const trimmedSearchTerm = searchTerm.trim().toLowerCase(); const trimmedSearchTerm = searchTerm.trim().toLowerCase();
const matchStrings = [result.name.toLowerCase(), result.type.toLowerCase(), result.description.toLowerCase()]; const matchStrings = [
result.name.toLowerCase(),
result.type.toLowerCase().replaceAll('_', ' '),
result.description.toLowerCase(),
];
if (result.type === 'spandrel_image_to_image') { if (result.type === 'spandrel_image_to_image') {
matchStrings.push('upscale'); matchStrings.push('upscale');
matchStrings.push('post-processing');
matchStrings.push('postprocessing');
matchStrings.push('post processing');
} }
return matchStrings.some((matchString) => matchString.includes(trimmedSearchTerm)); return matchStrings.some((matchString) => matchString.includes(trimmedSearchTerm));
}); });