diff --git a/components/CreatePanel.tsx b/components/CreatePanel.tsx index c28c01a..7000c68 100644 --- a/components/CreatePanel.tsx +++ b/components/CreatePanel.tsx @@ -284,6 +284,10 @@ export const CreatePanel: React.FC = ({ { id: 'acestep-v15-turbo-shift1', name: 'acestep-v15-turbo-shift1' }, { id: 'acestep-v15-turbo-shift3', name: 'acestep-v15-turbo-shift3' }, { id: 'acestep-v15-turbo-continuous', name: 'acestep-v15-turbo-continuous' }, + { id: 'acestep-v15-xl-turbo', name: 'acestep-v15-xl-turbo' }, + { id: 'acestep-v15-xl-sftturbo50', name: 'acestep-v15-xl-sftturbo50' }, + { id: 'acestep-v15-xl-sft', name: 'acestep-v15-xl-sft' }, + { id: 'acestep-v15-xl-base', name: 'acestep-v15-xl-base' }, ]; }, [fetchedModels]); @@ -296,6 +300,10 @@ export const CreatePanel: React.FC = ({ 'acestep-v15-turbo-shift3': '1.5TS3', 'acestep-v15-turbo-continuous': '1.5TC', 'acestep-v15-turbo': '1.5T', + 'acestep-v15-xl-turbo': '1.5XL-T', + 'acestep-v15-xl-sftturbo50': '1.5XL-ST50', + 'acestep-v15-xl-sft': '1.5XL-S', + 'acestep-v15-xl-base': '1.5XL-B', }; return mapping[modelId] || modelId; }; @@ -312,7 +320,7 @@ export const CreatePanel: React.FC = ({ // Check if model is the base variant (required for lego) const isBaseModel = (modelId: string): boolean => { - return modelId.startsWith('acestep-v15-base'); + return modelId.includes('acestep-v15-base') || modelId.includes('acestep-v15-xl-base'); }; // SFT model download/availability state for repaint mode diff --git a/server/src/routes/models.ts b/server/src/routes/models.ts index 07d9769..b5b9a2e 100644 --- a/server/src/routes/models.ts +++ b/server/src/routes/models.ts @@ -66,7 +66,12 @@ function buildCatalog(): ModelFile[] { { key: 'turbo-shift1', label: 'Turbo Shift-1', essential: false }, { key: 'turbo-shift3', label: 'Turbo Shift-3', essential: false }, { key: 'turbo-continuous', label: 'Turbo Continuous',essential: false }, + { key: 'xl-turbo', label: 'XL Turbo', essential: false }, + { key: 'xl-sft', label: 'XL SFT', essential: false }, + { key: 'xl-sftturbo50', label: 'XL SFT/Turbo 50/50', essential: false }, + { key: 'xl-base', label: 'XL Base', essential: false }, ]; + for (const v of ditVariants) { for (const q of QUANTS_DIT) { files.push({ @@ -76,7 +81,9 @@ function buildCatalog(): ModelFile[] { quant: q, variant: v.key, essential: v.essential && q === 'Q8_0', - approxSizeMB: q === 'BF16' ? 5200 : q === 'Q8_0' ? 2900 : q === 'Q6_K' ? 2200 : q === 'Q5_K_M' ? 1800 : 1500, + approxSizeMB: v.key.startsWith('xl') + ? (q === 'BF16' ? 10000 : q === 'Q8_0' ? 5500 : q === 'Q6_K' ? 4100 : q === 'Q5_K_M' ? 3500 : 3000) + : (q === 'BF16' ? 5200 : q === 'Q8_0' ? 2900 : q === 'Q6_K' ? 2200 : q === 'Q5_K_M' ? 1800 : 1500), }); } }