Treat upscale categories as Upscaler for auto base model resolution

This commit is contained in:
Dariusz L
2026-08-25 03:26:03 +02:00
parent c5371e7165
commit f358df46ae
2 changed files with 10 additions and 3 deletions
+3 -2
View File
@@ -4120,7 +4120,7 @@ test('base model alias resolves FLUX KREA as Flux.1 Krea', () => {
assert.equal(resolveBaseModelAlias.call(dialog, 'KREA'), 'Krea 2');
});
test('auto base model uses Any model for standalone SAM and Ultralytics models', () => {
test('auto base model uses Any model for standalone and upscale models', () => {
const getBaseModelIndependentSearchType = eval(`(${extractMethod(searchPanelMethodsSource, 'getBaseModelIndependentSearchType')})`);
const getMissingAutoBaseModelInfo = eval(`(${extractMethod(searchPanelMethodsSource, 'getMissingAutoBaseModelInfo')})`);
const getMissingAutoBaseModel = eval(`(${extractMethod(searchPanelMethodsSource, 'getMissingAutoBaseModel')})`);
@@ -4146,7 +4146,8 @@ test('auto base model uses Any model for standalone SAM and Ultralytics models',
for (const missing of [
{ category: 'sams', node_type: 'SAMLoader' },
{ category: 'ultralytics', node_type: 'UltralyticsDetectorProvider' }
{ category: 'ultralytics', node_type: 'UltralyticsDetectorProvider' },
{ category: 'UPSCALE_MODELS', node_type: 'UpscaleModelLoader' }
]) {
assert.equal(getMissingAutoBaseModel.call(dialog, missing), '');
assert.equal(getSearchBaseModelLabel.call(dialog, 'auto', missing), 'Auto (Any model)');
+7 -1
View File
@@ -84,7 +84,13 @@ export const searchPanelMethods = {
ultralytics: 'Ultralytics',
ultralytics_bbox: 'Ultralytics',
ultralytics_segm: 'Ultralytics',
yolo: 'Ultralytics'
yolo: 'Ultralytics',
upscale: 'Upscaler',
upscaler: 'Upscaler',
upscale_model: 'Upscaler',
upscale_models: 'Upscaler',
latent_upscale_model: 'Upscaler',
latent_upscale_models: 'Upscaler'
};
const candidates = [
missing?.category,