From 2371288fedaae838a29f152374e740fade7aaac2 Mon Sep 17 00:00:00 2001 From: Chenlei Hu Date: Fri, 11 Apr 2025 13:44:14 -0400 Subject: [PATCH] [Bug] Fix widgets values migration (#3404) --- src/utils/litegraphUtil.ts | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/src/utils/litegraphUtil.ts b/src/utils/litegraphUtil.ts index 3c7cdcc67..48cc87456 100644 --- a/src/utils/litegraphUtil.ts +++ b/src/utils/litegraphUtil.ts @@ -97,8 +97,18 @@ export function migrateWidgetsValues( const originalWidgetsInputs = Object.values(inputDefs).filter( (input) => widgetNames.has(input.name) || input.forceInput ) + // Count the number of original widgets inputs. + const numOriginalWidgets = _.sum( + originalWidgetsInputs.map((input) => + // If the input has control, it will have 2 widgets. + input.control_after_generate || + ['seed', 'noise_seed'].includes(input.name) + ? 2 + : 1 + ) + ) - if (originalWidgetsInputs.length === widgetsValues?.length) { + if (numOriginalWidgets === widgetsValues?.length) { return _.zip(originalWidgetsInputs, widgetsValues) .filter(([input]) => !input?.forceInput) .map(([_, value]) => value as TWidgetValue)