Add support for PRX Pixel T2I

This commit is contained in:
Jaret Burkett
2026-06-13 09:47:53 -06:00
parent a01c83073a
commit 6c3b82696e
9 changed files with 1464 additions and 0 deletions

View File

@@ -1004,6 +1004,27 @@ export const modelArchs: ModelArch[] = [
],
hasMultiLinePrompts: true,
},
{
name: 'prx_pixel',
label: 'PRXPixel (pixel space)',
group: 'image',
defaults: {
'config.process[0].model.name_or_path': ['Photoroom/prxpixel-t2i', defaultNameOrPath],
'config.process[0].model.quantize': [true, false],
'config.process[0].model.quantize_te': [true, false],
'config.process[0].train.timestep_type': ['linear', 'sigmoid'],
'config.process[0].network.conv': [undefined, 16],
'config.process[0].network.conv_alpha': [undefined, 16],
'config.process[0].model.low_vram': [true, false],
},
disableSections: [
'network.conv',
],
additionalSections: [
'model.low_vram',
'model.layer_offloading',
],
},
].sort((a, b) => {
// Sort by label, case-insensitive
return a.label.localeCompare(b.label, undefined, { sensitivity: 'base' });