diff --git a/tests/test_downloads_tab_workflow_route.mjs b/tests/test_downloads_tab_workflow_route.mjs index a2a6d79..37db101 100644 --- a/tests/test_downloads_tab_workflow_route.mjs +++ b/tests/test_downloads_tab_workflow_route.mjs @@ -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)'); diff --git a/web/resolver/search/search_panel.js b/web/resolver/search/search_panel.js index 83c906f..4c27761 100644 --- a/web/resolver/search/search_panel.js +++ b/web/resolver/search/search_panel.js @@ -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,