From 47e651225d1215df52fc5f72a4496797bfc32cc8 Mon Sep 17 00:00:00 2001 From: Eugene Brodsky Date: Mon, 26 Jun 2023 01:30:21 -0400 Subject: [PATCH] query for 'main' model type when populating UI lists to support renaming of 'pipeline' models to 'main' --- invokeai/frontend/web/src/app/components/App.tsx | 2 +- .../nodes/components/fields/ModelInputFieldComponent.tsx | 2 +- .../frontend/web/src/features/system/components/ModelSelect.tsx | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/invokeai/frontend/web/src/app/components/App.tsx b/invokeai/frontend/web/src/app/components/App.tsx index 3f1f2cf7a6..c93bd8791c 100644 --- a/invokeai/frontend/web/src/app/components/App.tsx +++ b/invokeai/frontend/web/src/app/components/App.tsx @@ -48,7 +48,7 @@ const App = ({ const isApplicationReady = useIsApplicationReady(); const { data: pipelineModels } = useListModelsQuery({ - model_type: 'pipeline', + model_type: 'main', }); const { data: controlnetModels } = useListModelsQuery({ model_type: 'controlnet', diff --git a/invokeai/frontend/web/src/features/nodes/components/fields/ModelInputFieldComponent.tsx b/invokeai/frontend/web/src/features/nodes/components/fields/ModelInputFieldComponent.tsx index 895d763882..741662655f 100644 --- a/invokeai/frontend/web/src/features/nodes/components/fields/ModelInputFieldComponent.tsx +++ b/invokeai/frontend/web/src/features/nodes/components/fields/ModelInputFieldComponent.tsx @@ -23,7 +23,7 @@ const ModelInputFieldComponent = ( const { t } = useTranslation(); const { data: pipelineModels } = useListModelsQuery({ - model_type: 'pipeline', + model_type: 'main', }); const data = useMemo(() => { diff --git a/invokeai/frontend/web/src/features/system/components/ModelSelect.tsx b/invokeai/frontend/web/src/features/system/components/ModelSelect.tsx index 916f70ef9a..f9eda624f2 100644 --- a/invokeai/frontend/web/src/features/system/components/ModelSelect.tsx +++ b/invokeai/frontend/web/src/features/system/components/ModelSelect.tsx @@ -24,7 +24,7 @@ const ModelSelect = () => { ); const { data: pipelineModels } = useListModelsQuery({ - model_type: 'pipeline', + model_type: 'main', }); const data = useMemo(() => {