@@ -29,13 +29,8 @@ import {
29
29
isVideoModelConfig ,
30
30
} from 'services/api/types' ;
31
31
32
- type ModelHookArgs = { excludeSubmodels ?: boolean } ;
33
-
34
32
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 ) =>
39
34
( filter : ( config : T ) => boolean = ( ) => true ) => {
40
35
const result = useGetModelConfigsQuery ( undefined ) ;
41
36
const modelConfigs = useMemo ( ( ) => {
@@ -45,7 +40,7 @@ const buildModelsHook =
45
40
46
41
return modelConfigsAdapterSelectors
47
42
. selectAll ( result . data )
48
- . filter ( ( config ) => typeGuard ( config , excludeSubmodels ) )
43
+ . filter ( ( config ) => typeGuard ( config ) )
49
44
. filter ( filter ) ;
50
45
} , [ filter , result . data ] ) ;
51
46
@@ -56,16 +51,12 @@ export const useRefinerModels = buildModelsHook(isRefinerMainModelModelConfig);
56
51
export const useLoRAModels = buildModelsHook ( isLoRAModelConfig ) ;
57
52
export const useControlLayerModels = buildModelsHook ( isControlLayerModelConfig ) ;
58
53
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 ) ( ) ;
63
56
export const useSpandrelImageToImageModels = buildModelsHook ( isSpandrelImageToImageModelConfig ) ;
64
57
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 ) ( ) ;
69
60
export const useGlobalReferenceImageModels = buildModelsHook (
70
61
( config ) =>
71
62
isIPAdapterModelConfig ( config ) ||
0 commit comments