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

exclude flux models from main model dropdown

Browse Source
This commit is contained in:
Mary Hipp 2024-08-20 20:17:16 -04:00 committed by Brandon Rising
parent c819da8859
commit ae9a1549ae
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/frontend/web/src/services/api/hooks/modelsByType.ts
View File

@ -9,7 +9,7 @@ import {
isFluxMainModelModelConfig, isFluxMainModelModelConfig,
isIPAdapterModelConfig, isIPAdapterModelConfig,
isLoRAModelConfig, isLoRAModelConfig,
isNonRefinerMainModelConfig, isNonRefinerNonFluxMainModelConfig,
isNonSDXLMainModelConfig, isNonSDXLMainModelConfig,
isRefinerMainModelModelConfig, isRefinerMainModelModelConfig,
isSDXLMainModelModelConfig, isSDXLMainModelModelConfig,
@ -35,7 +35,7 @@ const buildModelsHook =
return [modelConfigs, result] as const; return [modelConfigs, result] as const;
}; };
export const useMainModels = buildModelsHook(isNonRefinerMainModelConfig); export const useMainModels = buildModelsHook(isNonRefinerNonFluxMainModelConfig);
export const useNonSDXLMainModels = buildModelsHook(isNonSDXLMainModelConfig); export const useNonSDXLMainModels = buildModelsHook(isNonSDXLMainModelConfig);
export const useRefinerModels = buildModelsHook(isRefinerMainModelModelConfig); export const useRefinerModels = buildModelsHook(isRefinerMainModelModelConfig);
export const useFluxModels = buildModelsHook(isFluxMainModelModelConfig); export const useFluxModels = buildModelsHook(isFluxMainModelModelConfig);

4
invokeai/frontend/web/src/services/api/types.ts
View File

@ -126,6 +126,10 @@ export const isNonRefinerMainModelConfig = (config: AnyModelConfig): config is M
return config.type === 'main' && config.base !== 'sdxl-refiner'; return config.type === 'main' && config.base !== 'sdxl-refiner';
}; };
export const isNonRefinerNonFluxMainModelConfig = (config: AnyModelConfig): config is MainModelConfig => {
return config.type === 'main' && config.base !== 'sdxl-refiner' && config.base !== 'flux';
};
export const isRefinerMainModelModelConfig = (config: AnyModelConfig): config is MainModelConfig => { export const isRefinerMainModelModelConfig = (config: AnyModelConfig): config is MainModelConfig => {
return config.type === 'main' && config.base === 'sdxl-refiner'; return config.type === 'main' && config.base === 'sdxl-refiner';
}; };