Skip to content

Commit e348105

Browse files
feat(ui): add unknown model base support in ui
1 parent a87fcfd commit e348105

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

invokeai/frontend/web/src/features/modelManagerV2/models.ts

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -148,6 +148,7 @@ export const MODEL_BASE_TO_COLOR: Record<BaseModelType, string> = {
148148
'gemini-2.5': 'pink',
149149
veo3: 'purple',
150150
runway: 'green',
151+
unknown: 'red',
151152
};
152153

153154
/**
@@ -169,6 +170,7 @@ export const MODEL_BASE_TO_LONG_NAME: Record<BaseModelType, string> = {
169170
'gemini-2.5': 'Gemini 2.5',
170171
veo3: 'Veo3',
171172
runway: 'Runway',
173+
unknown: 'Unknown',
172174
};
173175

174176
/**
@@ -190,6 +192,7 @@ export const MODEL_BASE_TO_SHORT_NAME: Record<BaseModelType, string> = {
190192
'gemini-2.5': 'Gemini 2.5',
191193
veo3: 'Veo3',
192194
runway: 'Runway',
195+
unknown: 'Unknown',
193196
};
194197

195198
/**

invokeai/frontend/web/src/features/nodes/types/common.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -89,6 +89,7 @@ export const zBaseModelType = z.enum([
8989
'gemini-2.5',
9090
'veo3',
9191
'runway',
92+
'unknown',
9293
]);
9394
export type BaseModelType = z.infer<typeof zBaseModelType>;
9495
export const zMainModelBase = z.enum([

0 commit comments

Comments
 (0)