Spaces:
Running
Running
Nicholas Celestin commited on
Commit ·
351d046
1
Parent(s): 3f22414
Build update — 2026-05-22T18:36:38.568Z
Browse files
features/upscaler/model-registry.js
CHANGED
|
@@ -41,64 +41,6 @@ export const UPSCALER_RESAMPLER_MODELS = [
|
|
| 41 |
}
|
| 42 |
];
|
| 43 |
|
| 44 |
-
/**
|
| 45 |
-
* Shared model definitions for the image and video upscaler features.
|
| 46 |
-
* Single source of truth — all model <select> elements render from this list.
|
| 47 |
-
*/
|
| 48 |
-
|
| 49 |
-
export const UPSCALER_MODELS = [
|
| 50 |
-
{ url: 'models/4x-UpdraftSmall.onnx', scale: 4, label: 'Updraft Small (Custom)', sizeMB: 1.4, multipleOf: 32 },
|
| 51 |
-
{ url: 'models/4x-ClearRealityV1.onnx', scale: 4, label: 'ClearReality (SPAN)', sizeMB: 1.9 },
|
| 52 |
-
{ url: 'models/DAT_light_x4_dyn_OTF_4.onnx', scale: 4, label: 'DAT Light Restore (DAT-Light OTF)', sizeMB: 5 },
|
| 53 |
-
{ url: 'models/4x-UltraSharpV2_Lite.onnx', scale: 4, label: 'UltraSharp V2 Lite (RealPLKSR)', sizeMB: 30 },
|
| 54 |
-
{ url: 'models/4x-UltraSharpV2.onnx', scale: 4, label: 'UltraSharp V2 (DAT)', sizeMB: 52 },
|
| 55 |
-
// { url: 'models/super.onnx', scale: 4, label: 'Apple Super 188k', sizeMB: 5.5, multipleOf: 32 },
|
| 56 |
-
// { url: 'models/super_2.onnx', scale: 4, label: 'Apple Super 2 202k', sizeMB: 5.5, multipleOf: 32 },
|
| 57 |
-
// { url: 'models/super_3.onnx', scale: 4, label: 'Apple Super 3 244k', sizeMB: 5.5, multipleOf: 32 },
|
| 58 |
-
// { url: 'models/super_4.onnx', scale: 4, label: 'Apple Super 4 308k', sizeMB: 5.5, multipleOf: 32 },
|
| 59 |
-
// { url: 'models/super_5.onnx', scale: 4, label: 'Apple Super 5 358k', sizeMB: 5.5, multipleOf: 32 },
|
| 60 |
-
// { url: 'models/super_6.onnx', scale: 4, label: 'Apple Super 6 416k', sizeMB: 5.5, multipleOf: 32 },
|
| 61 |
-
// { url: 'models/super_7.onnx', scale: 4, label: 'Apple Super 7 450k', sizeMB: 5.5, multipleOf: 32 },
|
| 62 |
-
// { url: 'models/super_8.onnx', scale: 4, label: 'Apple Super 8 488k', sizeMB: 5.5, multipleOf: 32 },
|
| 63 |
-
// { url: 'models/super_9.onnx', scale: 4, label: 'Apple Super 9 521k', sizeMB: 5.5, multipleOf: 32 },
|
| 64 |
-
// { url: 'models/super_10.onnx', scale: 4, label: 'Apple Super 10 570k', sizeMB: 5.5, multipleOf: 32 },
|
| 65 |
-
// { url: 'models/super_11.onnx', scale: 4, label: 'Apple Super 11 617k', sizeMB: 5.5, multipleOf: 32 },
|
| 66 |
-
|
| 67 |
-
|
| 68 |
-
|
| 69 |
-
|
| 70 |
-
|
| 71 |
-
|
| 72 |
-
|
| 73 |
-
|
| 74 |
-
|
| 75 |
-
|
| 76 |
-
{
|
| 77 |
-
url: 'models/tinysr_fused.onnx',
|
| 78 |
-
scale: 4,
|
| 79 |
-
label: 'TinySR (DiT refiner)',
|
| 80 |
-
sizeMB: 687,
|
| 81 |
-
multipleOf: 128, // 128 LR × 4 = 512 HR (the fixed model input)
|
| 82 |
-
maxTileSize: 128, // same — every tile pads/crops to exactly 128 LR
|
| 83 |
-
precision: 'fp16',
|
| 84 |
-
upscaleBefore: true,
|
| 85 |
-
tileBlend: 'gaussian', // diffusion-style: hard-overlap shows seams
|
| 86 |
-
}
|
| 87 |
-
|
| 88 |
-
];
|
| 89 |
-
|
| 90 |
-
export const UPSCALER_RESAMPLER_MODELS = [
|
| 91 |
-
{ url: 'builtin:lanczos-4x', scale: 4, label: 'Lanczos' },
|
| 92 |
-
{ url: 'builtin:bicubic-4x', scale: 4, label: 'Bicubic' },
|
| 93 |
-
];
|
| 94 |
-
|
| 95 |
-
/**
|
| 96 |
-
* Render <option> elements for a model <select>.
|
| 97 |
-
* @param {typeof UPSCALER_MODELS} [models]
|
| 98 |
-
* @param {{ selected?: string, includeResamplers?: boolean }} [opts]
|
| 99 |
-
* - `selected` is matched against model URL
|
| 100 |
-
* - `includeResamplers` appends built-in non-ONNX upscale methods
|
| 101 |
-
*/
|
| 102 |
export function modelOptionsHTML(models = UPSCALER_MODELS, { selected, includeResamplers = false } = {}) {
|
| 103 |
const modelList = includeResamplers
|
| 104 |
? [...models, ...UPSCALER_RESAMPLER_MODELS]
|
|
@@ -132,4 +74,3 @@ export function modelOptionsHTML(models = UPSCALER_MODELS, { selected, includeRe
|
|
| 132 |
return `<option ${attrs.join(' ')}>${m.label}${sizeStr}</option>`;
|
| 133 |
}).join('\n ');
|
| 134 |
}
|
| 135 |
-
|
|
|
|
| 41 |
}
|
| 42 |
];
|
| 43 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 44 |
export function modelOptionsHTML(models = UPSCALER_MODELS, { selected, includeResamplers = false } = {}) {
|
| 45 |
const modelList = includeResamplers
|
| 46 |
? [...models, ...UPSCALER_RESAMPLER_MODELS]
|
|
|
|
| 74 |
return `<option ${attrs.join(' ')}>${m.label}${sizeStr}</option>`;
|
| 75 |
}).join('\n ');
|
| 76 |
}
|
|
|