fix(ui): fix metadata retrieval when has controlnet

This commit is contained in:
psychedelicious 2023-08-31 11:20:18 +10:00
parent 83163ddd9a
commit 8e470f9b6f

View File

@ -280,7 +280,7 @@ export type ConditioningInputFieldValue = z.infer<
export const zControlNetModel = zModelIdentifier; export const zControlNetModel = zModelIdentifier;
export type ControlNetModel = z.infer<typeof zControlNetModel>; export type ControlNetModel = z.infer<typeof zControlNetModel>;
export const zControlField = zInputFieldValueBase.extend({ export const zControlField = z.object({
image: zImageField, image: zImageField,
control_model: zControlNetModel, control_model: zControlNetModel,
control_weight: z.union([z.number(), z.array(z.number())]).optional(), control_weight: z.union([z.number(), z.array(z.number())]).optional(),
@ -295,11 +295,11 @@ export const zControlField = zInputFieldValueBase.extend({
}); });
export type ControlField = z.infer<typeof zControlField>; export type ControlField = z.infer<typeof zControlField>;
export const zControlInputFieldTemplate = zInputFieldValueBase.extend({ export const zControlInputFieldValue = zInputFieldValueBase.extend({
type: z.literal('ControlField'), type: z.literal('ControlField'),
value: zControlField.optional(), value: zControlField.optional(),
}); });
export type ControlInputFieldValue = z.infer<typeof zControlInputFieldTemplate>; export type ControlInputFieldValue = z.infer<typeof zControlInputFieldValue>;
export const zModelType = z.enum([ export const zModelType = z.enum([
'onnx', 'onnx',
@ -492,7 +492,7 @@ export const zInputFieldValue = z.discriminatedUnion('type', [
zUNetInputFieldValue, zUNetInputFieldValue,
zClipInputFieldValue, zClipInputFieldValue,
zVaeInputFieldValue, zVaeInputFieldValue,
zControlInputFieldTemplate, zControlInputFieldValue,
zEnumInputFieldValue, zEnumInputFieldValue,
zMainModelInputFieldValue, zMainModelInputFieldValue,
zSDXLMainModelInputFieldValue, zSDXLMainModelInputFieldValue,