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

Add model_type to the model state object

Browse Source
This commit is contained in:
Brandon Rising
2023-07-18 22:40:27 -04:00
parent e201ad2f51
commit 487455ef2e
8 changed files with 37 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
invokeai/frontend/web/src/features/parameters/store/generationSlice.ts
View File

@ -260,6 +260,7 @@ export const generationSlice = createSlice({
id: defaultModel,
name: model_name,
base_model,
model_type,
});
}
});

9
invokeai/frontend/web/src/features/parameters/store/parameterZodSchemas.ts
View File

@ -127,6 +127,14 @@ export const isValidHeight = (val: unknown): val is HeightParam =>
zHeight.safeParse(val).success;
const zBaseModel = z.enum(['sd-1', 'sd-2']);
const zModelType = z.enum([
'vae',
'lora',
'onnx',
'main',
'controlnet',
'embedding',
]);
export type BaseModelParam = z.infer<typeof zBaseModel>;
@ -137,6 +145,7 @@ export type BaseModelParam = z.infer<typeof zBaseModel>;
export const zMainModel = z.object({
model_name: z.string(),
base_model: zBaseModel,
model_type: zModelType,
});
/**