github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

Add CogView4 to frontend.

Browse Source
This commit is contained in:
Ryan Dick
2025-03-06 18:30:51 +00:00
committed by psychedelicious
parent e1133bc53f
commit f4e00ab261
11 changed files with 136 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -11,6 +11,7 @@ import type { AnyModelConfig } from 'services/api/types';
import {
isCLIPEmbedModelConfig,
isCLIPVisionModelConfig,
isCogView4MainModelModelConfig,
isControlLayerModelConfig,
isControlLoRAModelConfig,
isControlNetModelConfig,
@@ -61,6 +62,7 @@ export const useNonSDXLMainModels = buildModelsHook(isNonSDXLMainModelConfig);
export const useRefinerModels = buildModelsHook(isRefinerMainModelModelConfig);
export const useFluxModels = buildModelsHook(isFluxMainModelModelConfig);
export const useSD3Models = buildModelsHook(isSD3MainModelModelConfig);
export const useCogView4Models = buildModelsHook(isCogView4MainModelModelConfig);
export const useSDXLModels = buildModelsHook(isSDXLMainModelModelConfig);
export const useLoRAModels = buildModelsHook(isLoRAModelConfig);
export const useControlLoRAModel = buildModelsHook(isControlLoRAModelConfig);

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

@@ -247,6 +247,10 @@ export const isSD3MainModelModelConfig = (config: AnyModelConfig): config is Mai
return config.type === 'main' && config.base === 'sd-3';
};
export const isCogView4MainModelModelConfig = (config: AnyModelConfig): config is MainModelConfig => {
return config.type === 'main' && config.base === 'cogview4';
};
export const isFluxMainModelModelConfig = (config: AnyModelConfig): config is MainModelConfig => {
return config.type === 'main' && config.base === 'flux';
};