Remove changes to v1 workflow

This commit is contained in:
Brandon Rising 2024-08-12 14:06:17 -04:00 committed by Brandon
parent 5f59a828f9
commit 1d8545a76c
2 changed files with 1 additions and 13 deletions

View File

@ -114,11 +114,6 @@ const FIELD_TYPE_V1_TO_STATEFUL_FIELD_TYPE_V2: {
isCollection: false,
isCollectionOrScalar: false,
},
FluxMainModelField: {
name: 'FluxMainModelField',
isCollection: false,
isCollectionOrScalar: false,
},
SDXLMainModelField: {
name: 'SDXLMainModelField',
isCollection: false,

View File

@ -27,7 +27,7 @@ const zScheduler = z.enum([
'kdpm_2_a',
'lcm',
]);
const zBaseModel = z.enum(['any', 'sd-1', 'sd-2', 'sdxl', 'sdxl-refiner', 'flux']);
const zBaseModel = z.enum(['any', 'sd-1', 'sd-2', 'sdxl', 'sdxl-refiner']);
const zMainModel = z.object({
model_name: z.string().min(1),
base_model: zBaseModel,
@ -89,7 +89,6 @@ const zFieldTypeV1 = z.enum([
'ONNXModelField',
'Scheduler',
'SDXLMainModelField',
'FluxMainModelField',
'SDXLRefinerModelField',
'string',
'StringCollection',
@ -418,11 +417,6 @@ const zSDXLMainModelInputFieldValue = zInputFieldValueBase.extend({
value: zMainOrOnnxModel.optional(),
});
const zFluxMainModelInputFieldValue = zInputFieldValueBase.extend({
type: z.literal('FluxMainModelField'),
value: zMainModel.optional(),
});
const zSDXLRefinerModelInputFieldValue = zInputFieldValueBase.extend({
type: z.literal('SDXLRefinerModelField'),
value: zMainOrOnnxModel.optional(), // TODO: should narrow this down to a refiner model
@ -578,7 +572,6 @@ const zInputFieldValue = z.discriminatedUnion('type', [
zMainModelInputFieldValue,
zSchedulerInputFieldValue,
zSDXLMainModelInputFieldValue,
zFluxMainModelInputFieldValue,
zSDXLRefinerModelInputFieldValue,
zStringCollectionInputFieldValue,
zStringPolymorphicInputFieldValue,