fix refiner metadata

This commit is contained in:
Mary Hipp 2024-03-19 16:35:03 -04:00 committed by psychedelicious
parent 74a51571a0
commit 4e245e9331
3 changed files with 6 additions and 3 deletions

View File

@ -957,6 +957,7 @@
"refiner": "Refiner", "refiner": "Refiner",
"refinermodel": "Refiner Model", "refinermodel": "Refiner Model",
"refinerStart": "Refiner Start", "refinerStart": "Refiner Start",
"refinerSteps": "Refiner Steps",
"scheduler": "Scheduler", "scheduler": "Scheduler",
"steps": "Steps" "steps": "Steps"
}, },

View File

@ -247,10 +247,11 @@ export const handlers = {
recaller: recallers.refinerCFGScale, recaller: recallers.refinerCFGScale,
}), }),
refinerModel: buildHandlers({ refinerModel: buildHandlers({
getLabel: () => t('sdxl.refinerModel'), getLabel: () => t('sdxl.refinermodel'),
parser: parsers.refinerModel, parser: parsers.refinerModel,
recaller: recallers.refinerModel, recaller: recallers.refinerModel,
validator: validators.refinerModel, validator: validators.refinerModel,
renderValue: renderModelConfigValue,
}), }),
refinerNegativeAestheticScore: buildHandlers({ refinerNegativeAestheticScore: buildHandlers({
getLabel: () => t('sdxl.posAestheticScore'), getLabel: () => t('sdxl.posAestheticScore'),
@ -268,12 +269,12 @@ export const handlers = {
recaller: recallers.refinerScheduler, recaller: recallers.refinerScheduler,
}), }),
refinerStart: buildHandlers({ refinerStart: buildHandlers({
getLabel: () => t('sdxl.refiner_start'), getLabel: () => t('sdxl.refinerStart'),
parser: parsers.refinerStart, parser: parsers.refinerStart,
recaller: recallers.refinerStart, recaller: recallers.refinerStart,
}), }),
refinerSteps: buildHandlers({ refinerSteps: buildHandlers({
getLabel: () => t('sdxl.refiner_steps'), getLabel: () => t('sdxl.refinerSteps'),
parser: parsers.refinerSteps, parser: parsers.refinerSteps,
recaller: recallers.refinerSteps, recaller: recallers.refinerSteps,
}), }),

View File

@ -185,6 +185,7 @@ const parseRefinerModel: MetadataParseFunc<ParameterSDXLRefinerModel> = async (m
const key = await getModelKey(refiner_model, 'main'); const key = await getModelKey(refiner_model, 'main');
const refinerModelConfig = await fetchModelConfigWithTypeGuard(key, isRefinerMainModelModelConfig); const refinerModelConfig = await fetchModelConfigWithTypeGuard(key, isRefinerMainModelModelConfig);
const modelIdentifier = zModelIdentifierField.parse(refinerModelConfig); const modelIdentifier = zModelIdentifierField.parse(refinerModelConfig);
console.log({ modelIdentifier });
return modelIdentifier; return modelIdentifier;
}; };