fix(ui): ref image model types again

This commit is contained in:
psychedelicious
2025-07-04 20:31:44 +10:00
parent b67eb4134d
commit 7fac833813
4 changed files with 35 additions and 11 deletions

View File

@@ -69,6 +69,8 @@ export type SigLipModelConfig = S['SigLIPConfig'];
export type FLUXReduxModelConfig = S['FluxReduxConfig'];
export type ApiModelConfig = S['ApiModelConfig'];
export type MainModelConfig = DiffusersModelConfig | CheckpointModelConfig | ApiModelConfig;
export type FLUXKontextModelConfig = MainModelConfig;
export type ChatGPT4oModelConfig = ApiModelConfig;
export type AnyModelConfig =
| ControlLoRAModelConfig
| LoRAModelConfig
@@ -230,7 +232,7 @@ export const isFluxReduxModelConfig = (config: AnyModelConfig): config is FLUXRe
return config.type === 'flux_redux';
};
export const isChatGPT4oModelConfig = (config: AnyModelConfig): config is ApiModelConfig => {
export const isChatGPT4oModelConfig = (config: AnyModelConfig): config is ChatGPT4oModelConfig => {
return config.type === 'main' && config.base === 'chatgpt-4o';
};
@@ -246,7 +248,7 @@ export const isFluxKontextApiModelConfig = (config: AnyModelConfig): config is A
return config.type === 'main' && config.base === 'flux-kontext';
};
export const isFluxKontextModelConfig = (config: AnyModelConfig): config is MainModelConfig => {
export const isFluxKontextModelConfig = (config: AnyModelConfig): config is FLUXKontextModelConfig => {
return config.type === 'main' && config.base === 'flux' && config.name?.toLowerCase().includes('kontext');
};