Skip to content

Commit fa47e23

Browse files
refactor(ui): remove unused excludeSubmodels
I can't remember what this was for and don't see any reference to it. Maybe it's just remnants from a previous implementation?
1 parent bd893cf commit fa47e23

File tree

1 file changed

+6
-15
lines changed

1 file changed

+6
-15
lines changed

invokeai/frontend/web/src/services/api/hooks/modelsByType.ts

Lines changed: 6 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -29,13 +29,8 @@ import {
2929
isVideoModelConfig,
3030
} from 'services/api/types';
3131

32-
type ModelHookArgs = { excludeSubmodels?: boolean };
33-
3432
const buildModelsHook =
35-
<T extends AnyModelConfig>(
36-
typeGuard: (config: AnyModelConfig, excludeSubmodels?: boolean) => config is T,
37-
excludeSubmodels?: boolean
38-
) =>
33+
<T extends AnyModelConfig>(typeGuard: (config: AnyModelConfig) => config is T) =>
3934
(filter: (config: T) => boolean = () => true) => {
4035
const result = useGetModelConfigsQuery(undefined);
4136
const modelConfigs = useMemo(() => {
@@ -45,7 +40,7 @@ const buildModelsHook =
4540

4641
return modelConfigsAdapterSelectors
4742
.selectAll(result.data)
48-
.filter((config) => typeGuard(config, excludeSubmodels))
43+
.filter((config) => typeGuard(config))
4944
.filter(filter);
5045
}, [filter, result.data]);
5146

@@ -56,16 +51,12 @@ export const useRefinerModels = buildModelsHook(isRefinerMainModelModelConfig);
5651
export const useLoRAModels = buildModelsHook(isLoRAModelConfig);
5752
export const useControlLayerModels = buildModelsHook(isControlLayerModelConfig);
5853
export const useControlNetModels = buildModelsHook(isControlNetModelConfig);
59-
export const useT5EncoderModels = (args?: ModelHookArgs) =>
60-
buildModelsHook(isT5EncoderModelConfigOrSubmodel, args?.excludeSubmodels)();
61-
export const useCLIPEmbedModels = (args?: ModelHookArgs) =>
62-
buildModelsHook(isCLIPEmbedModelConfigOrSubmodel, args?.excludeSubmodels)();
54+
export const useT5EncoderModels = () => buildModelsHook(isT5EncoderModelConfigOrSubmodel)();
55+
export const useCLIPEmbedModels = () => buildModelsHook(isCLIPEmbedModelConfigOrSubmodel)();
6356
export const useSpandrelImageToImageModels = buildModelsHook(isSpandrelImageToImageModelConfig);
6457
export const useEmbeddingModels = buildModelsHook(isTIModelConfig);
65-
export const useVAEModels = (args?: ModelHookArgs) =>
66-
buildModelsHook(isVAEModelConfigOrSubmodel, args?.excludeSubmodels)();
67-
export const useFluxVAEModels = (args?: ModelHookArgs) =>
68-
buildModelsHook(isFluxVAEModelConfig, args?.excludeSubmodels)();
58+
export const useVAEModels = () => buildModelsHook(isVAEModelConfigOrSubmodel)();
59+
export const useFluxVAEModels = () => buildModelsHook(isFluxVAEModelConfig)();
6960
export const useGlobalReferenceImageModels = buildModelsHook(
7061
(config) =>
7162
isIPAdapterModelConfig(config) ||

0 commit comments

Comments
 (0)