towsty/aigen
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Add an xAIGen SNOFS switch for Flux and Krea.

Browse Source 
It is the same LoRA card as the rest of the chain; the toggle only adds or removes that card so it no longer comes back after you turn it off.

Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
Towsty
2026-08-30 13:27:30 -05:00
co-authored by Cursor
parent 9f82377b52
commit bfe94e4bb1
5 changed files with 88 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pages/index.vue
+66 -7
View File
@@ -757,6 +757,23 @@
</div>
<p class="mt-2 text-xs text-zinc-500">Flux is Klein 9B. Krea is Krea 2 Turbo. Video stays MiniMax or LTX.</p>
<div class="max-w-md">
<label
v-if="v2ConceptLoraEnabled && imageV2ConceptFile"
class="mb-2 flex cursor-pointer items-center gap-2 text-xs text-zinc-400"
>
<span class="relative inline-flex h-5 w-9 shrink-0 items-center">
<input
type="checkbox"
class="peer sr-only"
role="switch"
:checked="imageV2SnofsOn"
@change="onImageV2SnofsToggle"
>
<span class="absolute inset-0 rounded-full bg-zinc-700 transition peer-checked:bg-amber-400" />
<span class="absolute left-0.5 top-0.5 h-4 w-4 rounded-full bg-white transition peer-checked:translate-x-4" />
</span>
SNOFS
</label>
<LoraSelect
v-model="imageV2LoraStack"
:names="imageV2LoraOptions"
@@ -2008,15 +2025,19 @@ import {
IMAGE_V2_REFINE_FACE_PROMPT,
IMAGE_V2_REFINE_HAND,
IMAGE_V2_REFINE_HEAVY,
IMAGE_V2_SNOFS_CLIP,
IMAGE_V2_SNOFS_LORA,
IMAGE_V2_SNOFS_MODEL,
IMAGE_V2_STEPS_DEFAULT,
clampImageV2Denoise,
clampImageV2Size,
defaultImageV2LoraStack,
imageV2ConceptLoraName,
isImageV2ConceptLora,
isImageV2ConsistencyLora,
mergeLegacyImageV2Loras,
parseImageV2Engine,
removeImageV2ConceptLora,
upsertImageV2Lora,
type ImageV2Engine,
type ImageV2GenerateAspect,
@@ -2365,8 +2386,18 @@ function imageV2EngineLoras(stack?: unknown) {
)
}
function imageV2ConceptName() {
return imageV2ConceptLoraName(
v2Engine.value,
String(runtimeConfig.public.kreaConceptLora || IMAGE_V2_KREA_CONCEPT_LORA || '').trim()
)
}
function seedImageV2LorasIfEmpty() {
if (imageV2LoraStack.value.length) return
try {
if (localStorage.getItem(imageV2LoraStoreKey(v2Engine.value)) != null) return
} catch { /* ignore */ }
imageV2LoraStack.value = defaultImageV2LoraStack({
engine: v2Engine.value,
mode: v2Mode.value,
@@ -2439,9 +2470,14 @@ const videoLoraOptions = computed(() => mergeLoraOptions(videoLoras.value, [
...extensionQueue.value.flatMap(item => (item.loraStack || []).map(entry => entry.name))
]))
const imageLoraOptions = computed(() => mergeLoraOptions(imageLoras.value, imageLoraStack.value.map(item => item.name)))
const imageV2ConceptFile = computed(() => imageV2ConceptName())
const imageV2SnofsOn = computed(() => imageV2LoraStack.value.some(item => isImageV2ConceptLora(item.name, v2Engine.value)))
const imageV2LoraOptions = computed(() => mergeLoraOptions(
filterLorasForImageEngine(imageLoras.value, v2Engine.value),
filterLorasForImageEngine(imageV2LoraStack.value.map(item => item.name), v2Engine.value)
filterLorasForImageEngine([
...imageV2LoraStack.value.map(item => item.name),
...(v2ConceptLoraEnabled.value && imageV2ConceptFile.value ? [imageV2ConceptFile.value] : [])
], v2Engine.value)
))
const identityRefs = ref<(File | null)[]>([null, null, null, null])
const identityPreviews = ref(['', '', '', ''])
@@ -3578,14 +3614,37 @@ function persistImageV2LoraStack(stack: LoraStackItem[], engine: ImageV2Engine)
} catch { /* ignore */ }
}
function readStoredLoraStackIfPresent(key: string) {
try {
if (localStorage.getItem(key) == null) return null
return readLoraStack(localStorage.getItem(key) || '')
} catch {
return null
}
}
function readImageV2LoraStack(engine: ImageV2Engine) {
const keyed = readStoredLoraStack(imageV2LoraStoreKey(engine))
if (keyed.length) return filterLoraStackForImageEngine(keyed, engine)
const legacy = readStoredLoraStack('aigen-image-v2-lora')
if (!legacy.length) return [] as LoraStackItem[]
const keyed = readStoredLoraStackIfPresent(imageV2LoraStoreKey(engine))
if (keyed) return filterLoraStackForImageEngine(keyed, engine)
const legacy = readStoredLoraStackIfPresent('aigen-image-v2-lora')
if (!legacy) return [] as LoraStackItem[]
return filterLoraStackForImageEngine(legacy, engine)
}
function setImageV2Snofs(on: boolean) {
const name = imageV2ConceptName()
if (!name) return
if (on) {
imageV2LoraStack.value = upsertImageV2Lora(imageV2LoraStack.value, name, IMAGE_V2_SNOFS_MODEL, IMAGE_V2_SNOFS_CLIP)
return
}
imageV2LoraStack.value = removeImageV2ConceptLora(imageV2LoraStack.value, v2Engine.value)
}
function onImageV2SnofsToggle(event: Event) {
setImageV2Snofs((event.target as HTMLInputElement).checked)
}
function mergeLoraOptions(listed: string[], extra: string[]) {
const names = [...listed]
for (const name of extra) {
@@ -5724,7 +5783,7 @@ function applyRefinePreset(kind: 'face' | 'hand' | 'heavy') {
if (kind === 'face') {
v2Denoise.value = IMAGE_V2_REFINE_FACE.strength
if (v2Engine.value === 'flux') {
if (v2ConceptLoraEnabled.value) {
if (v2ConceptLoraEnabled.value && imageV2SnofsOn.value) {
imageV2LoraStack.value = upsertImageV2Lora(imageV2LoraStack.value, IMAGE_V2_SNOFS_LORA, IMAGE_V2_REFINE_FACE.snofsModel, IMAGE_V2_REFINE_FACE.snofsClip)
}
imageV2LoraStack.value = upsertImageV2Lora(imageV2LoraStack.value, IMAGE_V2_CONSISTENCY_LORA, IMAGE_V2_REFINE_FACE.consistencyModel, IMAGE_V2_REFINE_FACE.consistencyClip)
@@ -5735,7 +5794,7 @@ function applyRefinePreset(kind: 'face' | 'hand' | 'heavy') {
if (kind === 'hand') {
v2Denoise.value = IMAGE_V2_REFINE_HAND.strength
if (v2Engine.value === 'flux') {
if (v2ConceptLoraEnabled.value) {
if (v2ConceptLoraEnabled.value && imageV2SnofsOn.value) {
imageV2LoraStack.value = upsertImageV2Lora(imageV2LoraStack.value, IMAGE_V2_SNOFS_LORA, IMAGE_V2_REFINE_HAND.snofsModel, IMAGE_V2_REFINE_HAND.snofsClip)
}
imageV2LoraStack.value = upsertImageV2Lora(imageV2LoraStack.value, IMAGE_V2_CONSISTENCY_LORA, IMAGE_V2_REFINE_HAND.consistencyModel, IMAGE_V2_REFINE_HAND.consistencyClip)
server/api/v2/generate.post.ts
+2 -4
View File
@@ -6,8 +6,6 @@ import {
IMAGE_V2_CFG_DEFAULT,
IMAGE_V2_CONSISTENCY_CLIP,
IMAGE_V2_CONSISTENCY_MODEL,
IMAGE_V2_SNOFS_CLIP,
IMAGE_V2_SNOFS_MODEL,
IMAGE_V2_STEPS_DEFAULT,
IMAGE_V2_TURBO_CFG,
IMAGE_V2_TURBO_STEPS,
@@ -221,8 +219,8 @@ export default defineEventHandler(async (event) => {
mode === 'generate' || engine === 'krea' ? 0 : IMAGE_V2_CONSISTENCY_CLIP
)
const conceptAllowed = isXaigenStudio()
const generateSnofsFallback = !conceptAllowed || engine === 'krea' ? 0 : IMAGE_V2_SNOFS_MODEL
const generateSnofsClipFallback = !conceptAllowed || engine === 'krea' ? 0 : IMAGE_V2_SNOFS_CLIP
const generateSnofsFallback = 0
const generateSnofsClipFallback = 0
await listStudioLoras()
const loraStack = assertImageV2LoraStack(parsePostedLoraStack(fields.loraStack || fields.lora, 'image'), engine)
server/utils/studioQueue.ts
+2 -2
View File
@@ -797,8 +797,8 @@ async function startStudioEditJob(item: StudioJob) {
steps: payload.steps,
seed: live.library?.seed || payload.seed,
cfg: payload.cfg,
snofsModel: payload.snofsModel ?? (isXaigenStudio() ? 0.65 : 0),
snofsClip: payload.snofsClip ?? (isXaigenStudio() ? 0.35 : 0),
snofsModel: payload.snofsModel ?? 0,
snofsClip: payload.snofsClip ?? 0,
consistencyModel: payload.consistencyModel ?? (mode === 'generate' || payload.engine === 'krea' ? 0 : 0.7),
consistencyClip: payload.consistencyClip ?? (mode === 'generate' || payload.engine === 'krea' ? 0 : 0.7),
megapixels: payload.scaleMegapixels ?? 1,
utils/imageV2.ts
+17 -4
View File
@@ -74,7 +74,7 @@ export const IMAGE_V2_KREA_UNET_CANDIDATES = [
'krea2_turbo_nvfp4.safetensors',
'krea2_turbo_fp8_scaled.safetensors'
] as const
export const IMAGE_V2_KREA_CONCEPT_LORA = ''
export const IMAGE_V2_KREA_CONCEPT_LORA = 'xaigen-snofs_krea_v1_3D.safetensors'
export const IMAGE_V2_ROLE_HEADERS: Record<Exclude<ImageV2Task, 'scene' | 'refine' | 't2i'>, string> = {
outfit: 'Person, face, body, pose, and background from image 1. Clothing only from image 2. Fit the outfit from image 2 to the body in image 1. Do not copy image 2’s face, body shape, or pose.',
@@ -175,9 +175,18 @@ export type ImageV2PresetSettings = {
height?: number
}
export function isImageV2ConceptLora(name: string) {
export function imageV2ConceptLoraName(engine: ImageV2Engine, kreaConcept?: string) {
if (engine === 'krea') return String(kreaConcept || IMAGE_V2_KREA_CONCEPT_LORA || '').trim()
return IMAGE_V2_SNOFS_LORA
}
export function isImageV2ConceptLora(name: string, engine?: ImageV2Engine) {
const id = loraIdentityKey(name)
return id === loraIdentityKey(IMAGE_V2_SNOFS_LORA) || /klein_snofs/i.test(id)
const flux = id === loraIdentityKey(IMAGE_V2_SNOFS_LORA) || /klein_snofs/i.test(id)
const krea = id === loraIdentityKey(IMAGE_V2_KREA_CONCEPT_LORA) || /snofs_krea/i.test(id)
if (engine === 'krea') return krea
if (engine === 'flux') return flux
return flux || krea
}
export function isImageV2ConsistencyLora(name: string) {
@@ -203,6 +212,10 @@ export function upsertImageV2Lora(stack: unknown, name: string, model: number, c
return [item, ...current]
}
export function removeImageV2ConceptLora(stack: unknown, engine?: ImageV2Engine) {
return normalizeLoraStack(stack).filter(item => !isImageV2ConceptLora(item.name, engine))
}
export function defaultImageV2LoraStack(opts: {
engine: ImageV2Engine
mode: ImageV2Mode
@@ -242,7 +255,7 @@ export function mergeLegacyImageV2Loras(stack: unknown, opts: {
const name = String(opts.kreaConcept || '').trim()
const model = Number(opts.snofsModel) || 0
const clip = Number(opts.snofsClip) || 0
if (name && (model > 0 || clip > 0) && !items.some(item => loraIdentityKey(item.name) === loraIdentityKey(name))) {
if (name && (model > 0 || clip > 0) && !items.some(item => isImageV2ConceptLora(item.name, 'krea'))) {
return upsertImageV2Lora(items, name, model, clip)
}
return items
utils/loras.ts
+1 -1
View File
@@ -31,7 +31,7 @@ export function loraIdentityKey(name: string) {
export function isXaigenOnlyLora(name: string) {
const base = loraBasename(name)
return /^xaigen/i.test(base) || /klein_snofs/i.test(base)
return /^xaigen/i.test(base) || /klein_snofs|snofs_krea/i.test(base)
}
/** Pick the real Comfy filename: exact path, then same basename ignoring `xaigen-`, then suffix/contains. */