diff --git a/docs/Features/Presets.md b/docs/Features/Presets.md index d59be0fdd..6fb45b50e 100644 --- a/docs/Features/Presets.md +++ b/docs/Features/Presets.md @@ -2,6 +2,10 @@ (TODO: Info about how presets work and all that) +Presets allow you to save and reuse sets of generation parameters across different workflows. When you create a new preset, you choose the parameters to be applied by that preset when you select it later. You can choose multiple presets, they "overlay" their settings. + +When you have presets selected, the parameters they include are locked from editing until you either unselect the presets or "apply" them to your prompt. + # Official SDXL Preset Pack These are the presets that were used on DreamStudio (Stability AI's web generation platform) around the time that SDXL was launched to the public in 2023. @@ -17,3 +21,32 @@ These are the presets that were used on DreamStudio (Stability AI's web generati - You can click on any preset to use it, or unclick it to disable it. - (Note: if it behaves weird, refresh the page. Alpha projects do silly things sometimes!) + +## Linking Models and LoRA to a Preset + +It's common for models and LoRA to have specific recommended values for CFG scale, steps, resolution, sampler, etc. Or, you may just have your own preferences for getting the "look" you want from them. + +To do this, create a preset with your preferred settings. Then, go to the *Edit Metadata* dialog for the model or LoRA and pick the preset from the dropdown and save. After that, any time you select that model or LoRA, the linked preset is also selected and can be applied. + +Some nuances of this feature: + +1. Presets linked to *models* are considered *mutually exclusive* to one another. So, if you select a model, its preset is selected, and if you click another model that has its own preset, the first model's preset is unselected and the new model's preset is selected. This avoids accidentally "stacking" incompatible presets for different models. (LoRA presets don't work this way -- they are just selected with the LoRA and are *not* automatically de-selected if you disable the LoRA.) + +2. You *can* have a single Preset linked to multiple models or LoRA, but you can only link one preset at a time to each model or LoRA. + +3. These links are *user-level* setting, not global. + +4. If a model's linked preset specifies that a given LoRA be included at a certain weight, and that LoRA *also* has a preset, that preset will also be selected. (This only nests one level -- LoRA presets can load other LoRA, but it won't select those other LoRAs' presets.) + +5. If you enable the User Setting `AutoApplyModelPresets`, rather than *selecting* the appropriate preset for the model and downstream LoRA, it will immediately *apply* those settings. This is very convenient if you just want to flip between models and have your preferred weights, steps, CFG scale, etc. be set instantly. + +6. If you enable the User Setting `IgnoreNestedZeroWeightLoraPresets`, the logic in (3) above is slightly adjusted, so any LoRA in the model's preset that has a weight of `0` will *not* have its own preset loaded. The reasoning for this option is as follows: + + - There are many situations where a LoRA is on the same platform as a model, but is incompatible with the model. For example, you shouldn't use an LCM LoRA with a model that already has LCM or DMD "built in." + - When flipping between models, it can be a chore to remember to add or remove these "process" LoRA that apply to some models but not others. + - Presets don't support allowing you to *remove* a LoRA. + - However, you *can add* a LoRA with a weight of `0`. The LoRA still takes up VRAM, but has no impact on the generation. + - So, if you keep accidentally leaving an LCM model active when you switch to a DMD model, you can set the DMD model's preset to "include" the LCM model, but with a weight of `0`. + - It's also likely you'd want to set a preset for the LCM LoRA, because when you *normally* add it, you want to override the usual step count, etc. + - But in this situation with the DMD model, you want the LCM LoRA to be effectively disabled *and* you wouldn't want its preset to override your DMD model's preset. + - If this is not a situation you regularly have to deal with, you can leave this option disabled; it's primarily for users who have lower-end devices who prefer to stick with low-step configurations regardless which model they are using. diff --git a/src/Core/Settings.cs b/src/Core/Settings.cs index 7fba6a087..eb41f9933 100644 --- a/src/Core/Settings.cs +++ b/src/Core/Settings.cs @@ -391,6 +391,12 @@ public class UserUIData : AutoConfiguration [ConfigComment("If enabled, shifting to next/previous image (eg with arrow keys) in history or batch view,\ncycles at the ends (jumps from the start to the end or vice versa).\nIf disabled, shifting will simply stop at the ends.\nIf 'only arrow keys', cycling happens when you press the arrow keys, but not other actions (eg deleting an image will not cycle).")] [ManualSettingsOptions(Vals = ["true", "false", "only_arrows"], ManualNames = ["Enabled", "Disabled", "Only Arrow Keys"])] public string ImageShiftingCycles = "true"; + + [ConfigComment("When a model with a linked preset is selected, should the preset parameters be applied immediately? By default, the preset is just selected for review.")] + public bool AutoApplyModelPresets = false; + + [ConfigComment("If enabled, when applying the model's preset that includes a LoRA with a weight of 0, the LoRA's own preset is ignored.")] + public bool IgnoreModelPresetZeroWeightLoraPresets = false; } [ConfigComment("Settings related to the user interface, entirely contained to the frontend.")] diff --git a/src/Pages/_Generate/GenTabModals.cshtml b/src/Pages/_Generate/GenTabModals.cshtml index d6875a0d0..d24d17f4b 100644 --- a/src/Pages/_Generate/GenTabModals.cshtml +++ b/src/Pages/_Generate/GenTabModals.cshtml @@ -156,6 +156,19 @@ Description:
+
+
+
+
Linked Preset
+ @WebUtil.TextPopoverButton("linkedpreset", "If set, this preset will be selected when you choose this model. Enable auto-apply in settings to instead apply it automatically.") +
+
+
+ + +
+
+
diff --git a/src/WebAPI/AdminAPI.cs b/src/WebAPI/AdminAPI.cs index 418e5fcde..7a3db2e9f 100644 --- a/src/WebAPI/AdminAPI.cs +++ b/src/WebAPI/AdminAPI.cs @@ -66,15 +66,15 @@ public static JObject AutoConfigToParamData(AutoConfiguration config, bool hideR { val = AutoConfigToParamData(subConf); } - if (data.Field.GetCustomAttribute() is not null) + if (hideRestricted && data.Field.GetCustomAttribute() is not null) { continue; } - if (hideRestricted && data.Field.GetCustomAttribute() is not null) + if (data.Field.GetCustomAttribute() is not null) { continue; } - string[] vals = data.Field.GetCustomAttribute()?.Options ?? null; + string[] vals = data.Field.GetCustomAttribute()?.Options ?? null; string[] val_names = null; if (vals is not null) { diff --git a/src/WebAPI/BasicAPIFeatures.cs b/src/WebAPI/BasicAPIFeatures.cs index 75ab8f16b..6537fa520 100644 --- a/src/WebAPI/BasicAPIFeatures.cs +++ b/src/WebAPI/BasicAPIFeatures.cs @@ -226,6 +226,20 @@ public static async Task InstallConfirmWS(Session session, WebSocket so public static async Task GetMyUserData(Session session) { Settings.User.AutoCompleteData settings = session.User.Settings.AutoComplete; + JObject itemPresetLinks = new JObject(); + try + { + string rawJson = session.User.GetGenericData("itempresetlinks", "data"); + if (!string.IsNullOrWhiteSpace(rawJson)) + { + itemPresetLinks = JObject.Parse(rawJson); + } + } + catch + { + // If JSON parsing fails, just return empty object + itemPresetLinks = new JObject(); + } return new JObject() { ["user_name"] = session.User.UserID, @@ -233,7 +247,8 @@ public static async Task GetMyUserData(Session session) ["language"] = session.User.Settings.Language, ["permissions"] = JArray.FromObject(session.User.GetPermissions()), ["starred_models"] = JObject.Parse(session.User.GetGenericData("starred_models", "full") ?? "{}"), - ["autocompletions"] = string.IsNullOrWhiteSpace(settings.Source) ? null : new JArray(AutoCompleteListHelper.GetData(settings.Source, settings.EscapeParens, settings.Suffix, settings.SpacingMode)) + ["autocompletions"] = string.IsNullOrWhiteSpace(settings.Source) ? null : new JArray(AutoCompleteListHelper.GetData(settings.Source, settings.EscapeParens, settings.Suffix, settings.SpacingMode)), + ["item_preset_links"] = itemPresetLinks }; } diff --git a/src/WebAPI/ModelsAPI.cs b/src/WebAPI/ModelsAPI.cs index 9dd73bef9..5b323e249 100644 --- a/src/WebAPI/ModelsAPI.cs +++ b/src/WebAPI/ModelsAPI.cs @@ -29,6 +29,8 @@ public static void Register() API.RegisterAPICall(TestPromptFill, false, Permissions.FundamentalModelAccess); API.RegisterAPICall(EditWildcard, true, Permissions.EditWildcards); API.RegisterAPICall(EditModelMetadata, true, Permissions.EditModelMetadata); + API.RegisterAPICall(SaveItemPresetLink, true, Permissions.ManagePresets); + API.RegisterAPICall(ClearItemPresetLink, true, Permissions.ManagePresets); API.RegisterAPICall(DoModelDownloadWS, true, Permissions.DownloadModels); API.RegisterAPICall(GetModelHash, true, Permissions.EditModelMetadata); API.RegisterAPICall(ForwardMetadataRequest, false, Permissions.EditModelMetadata); @@ -496,52 +498,52 @@ public static async Task EditModelMetadata(Session session, { return new JObject() { ["error"] = "Model not found." }; } - lock (handler.ModificationLock) - { - actualModel.Title = string.IsNullOrWhiteSpace(title) ? null : title; - actualModel.Description = description; - if (!string.IsNullOrWhiteSpace(type)) - { - actualModel.ModelClass = T2IModelClassSorter.ModelClasses.GetValueOrDefault(type); - } - if (standard_width > 0) - { - actualModel.StandardWidth = standard_width; - } - if (standard_height > 0) - { - actualModel.StandardHeight = standard_height; - } - actualModel.Metadata ??= new(); - if (!string.IsNullOrWhiteSpace(preview_image)) - { - if (preview_image == "clear") - { - actualModel.PreviewImage = "imgs/model_placeholder.jpg"; - actualModel.Metadata.PreviewImage = null; - } - else - { - ImageFile img = ImageFile.FromDataString(preview_image).ToMetadataJpg(preview_image_metadata); - if (img is not null) - { - actualModel.PreviewImage = img.AsDataString(); - actualModel.Metadata.PreviewImage = actualModel.PreviewImage; - } - } - } - actualModel.Metadata.Author = author; - actualModel.Metadata.UsageHint = usage_hint; - actualModel.Metadata.Date = date; - actualModel.Metadata.License = license; - actualModel.Metadata.TriggerPhrase = trigger_phrase; - actualModel.Metadata.Tags = string.IsNullOrWhiteSpace(tags) ? null : tags.Split(',', StringSplitOptions.TrimEntries | StringSplitOptions.RemoveEmptyEntries); - actualModel.Metadata.IsNegativeEmbedding = is_negative_embedding; - actualModel.Metadata.LoraDefaultWeight = lora_default_weight; - actualModel.Metadata.LoraDefaultConfinement = lora_default_confinement; - actualModel.Metadata.PredictionType = string.IsNullOrWhiteSpace(prediction_type) ? null : prediction_type; - } - handler.ResetMetadataFrom(actualModel); + lock (handler.ModificationLock) + { + actualModel.Title = string.IsNullOrWhiteSpace(title) ? null : title; + actualModel.Description = description; + if (!string.IsNullOrWhiteSpace(type)) + { + actualModel.ModelClass = T2IModelClassSorter.ModelClasses.GetValueOrDefault(type); + } + if (standard_width > 0) + { + actualModel.StandardWidth = standard_width; + } + if (standard_height > 0) + { + actualModel.StandardHeight = standard_height; + } + actualModel.Metadata ??= new(); + if (!string.IsNullOrWhiteSpace(preview_image)) + { + if (preview_image == "clear") + { + actualModel.PreviewImage = "imgs/model_placeholder.jpg"; + actualModel.Metadata.PreviewImage = null; + } + else + { + ImageFile img = ImageFile.FromDataString(preview_image).ToMetadataJpg(preview_image_metadata); + if (img is not null) + { + actualModel.PreviewImage = img.AsDataString(); + actualModel.Metadata.PreviewImage = actualModel.PreviewImage; + } + } + } + actualModel.Metadata.Author = author; + actualModel.Metadata.UsageHint = usage_hint; + actualModel.Metadata.Date = date; + actualModel.Metadata.License = license; + actualModel.Metadata.TriggerPhrase = trigger_phrase; + actualModel.Metadata.Tags = string.IsNullOrWhiteSpace(tags) ? null : tags.Split(',', StringSplitOptions.TrimEntries | StringSplitOptions.RemoveEmptyEntries); + actualModel.Metadata.IsNegativeEmbedding = is_negative_embedding; + actualModel.Metadata.LoraDefaultWeight = lora_default_weight; + actualModel.Metadata.LoraDefaultConfinement = lora_default_confinement; + actualModel.Metadata.PredictionType = string.IsNullOrWhiteSpace(prediction_type) ? null : prediction_type; + } + handler.ResetMetadataFrom(actualModel); _ = Utilities.RunCheckedTask(() => actualModel.ResaveModel(), "model resave"); Interlocked.Increment(ref ModelEditID); return new JObject() { ["success"] = true }; @@ -889,4 +891,62 @@ void doMoveNow(string oldPath) Interlocked.Increment(ref ModelEditID); return new JObject() { ["success"] = true }; } + + /// Saves a reference to a preset for a model or LoRA to the user's data. + /// The user session context. + /// The type of item: 'model' or 'lora'. + /// The name/identifier of the model or LoRA. + /// The title of the preset to link, or empty string to clear the link. + /// A JSON object containing the success status, the key that was saved/cleared, and the preset title. + public static async Task SaveItemPresetLink(Session session, string itemType, string itemName, string presetTitle) + { + string key = $"{itemType}:{itemName}"; + try + { + string rawJson = session.User.GetGenericData("itempresetlinks", "data") ?? "{}"; + JObject links = JObject.Parse(rawJson); + if (!string.IsNullOrWhiteSpace(presetTitle)) + { + links[key] = presetTitle; + } + else if (links.ContainsKey(key)) + { + links.Remove(key); + } + string newJson = links.ToString(Newtonsoft.Json.Formatting.None); + session.User.SaveGenericData("itempresetlinks", "data", newJson); + return new JObject() { ["success"] = true, ["key"] = key, ["preset"] = presetTitle }; + } + catch (Exception ex) + { + Logs.Error($"Error saving preset link: {ex.ReadableString()}"); + } + } + + /// Clears a reference to a preset for a model or LoRA from the user's data. + /// The user session context. + /// The type of item: 'model' or 'lora'. + /// The name/identifier of the model or LoRA. + /// A JSON object containing the success status and the key that was cleared. + public static async Task ClearItemPresetLink(Session session, string itemType, string itemName) + { + string key = $"{itemType}:{itemName}"; + try + { + string rawJson = session.User.GetGenericData("itempresetlinks", "data") ?? "{}"; + JObject links = JObject.Parse(rawJson); + if (links.ContainsKey(key)) + { + links.Remove(key); + string newJson = links.ToString(Newtonsoft.Json.Formatting.None); + session.User.SaveGenericData("itempresetlinks", "data", newJson); + } + return new JObject() { ["success"] = true, ["key"] = key }; + } + catch (Exception ex) + { + Logs.Error($"Error clearing preset link: {ex.Message}"); + return new JObject() { ["success"] = false, ["error"] = ex.Message }; + } + } } diff --git a/src/wwwroot/css/genpage.css b/src/wwwroot/css/genpage.css index 28072ac07..54f757795 100644 --- a/src/wwwroot/css/genpage.css +++ b/src/wwwroot/css/genpage.css @@ -559,6 +559,9 @@ body { font-size: 80%; color: var(--text-soft); } +.model_preset { + color: var(--text-soft); +} .refresh-button { margin: 0.5rem; background-color: var(--button-background); @@ -1660,3 +1663,6 @@ body { height: 1lh; padding-top: 0 !important; } +button.disabled-dim { + opacity: 0.5; +} diff --git a/src/wwwroot/js/genpage/gentab/loras.js b/src/wwwroot/js/genpage/gentab/loras.js index 845c2e76c..0fb8be90d 100644 --- a/src/wwwroot/js/genpage/gentab/loras.js +++ b/src/wwwroot/js/genpage/gentab/loras.js @@ -275,6 +275,8 @@ class LoraHelper { } else { this.selected.push(new SelectedLora(name, null, null, data)); + // Handle LoRA preset selection/application when adding a LoRA + selectOrApplyLoraPresetOnSelection(name); } this.rebuildParams(); this.rebuildUI(); diff --git a/src/wwwroot/js/genpage/gentab/models.js b/src/wwwroot/js/genpage/gentab/models.js index 6b2d8d7e6..9f3349eee 100644 --- a/src/wwwroot/js/genpage/gentab/models.js +++ b/src/wwwroot/js/genpage/gentab/models.js @@ -176,6 +176,8 @@ function editModel(model, browser) { getRequiredElementById('edit_model_lora_default_confinement').value = model.lora_default_confinement || ''; getRequiredElementById('edit_model_lora_default_confinement_div').style.display = model.architecture && model.architecture.endsWith('/lora') ? 'block' : 'none'; let run = () => { + buildModelOverridePresetInput(); + updatePresetLinkButtonState('edit_model_override_preset_id'); triggerChangeFor(modelsHelpers.enableImageElem); $('#edit_model_modal').modal('show'); }; @@ -265,10 +267,16 @@ function save_edit_model() { data['lora_default_weight'] = (model.architecture || '').endsWith('/lora') ? getRequiredElementById('edit_model_lora_default_weight').value : ''; data['lora_default_confinement'] = (model.architecture || '').endsWith('/lora') ? getRequiredElementById('edit_model_lora_default_confinement').value : ''; data.subtype = curModelMenuBrowser.subType; + function complete() { genericRequest('EditModelMetadata', data, data => { curModelMenuBrowser.browser.lightRefresh(); }); + // Save preset link using centralized function + let presetSelect = document.getElementById('edit_model_override_preset_id'); + let presetTitle = presetSelect?.value || ''; + // Determine item type based on browser subType (LoRA, Embedding, etc) or model architecture + saveItemPresetLink(curModelMenuBrowser.subType, model.name, presetTitle); $('#edit_model_modal').modal('hide'); } if (modelsHelpers.enableImageElem.checked) { @@ -645,7 +653,14 @@ class ModelBrowserWrapper { interject += `${getOptLine("Default LoRA Weight", model.data.lora_default_weight)}${getOptLine("Default LoRA Confinement", confinementName)}`; searchableAdded += `, Default LoRA Weight: ${model.data.lora_default_weight}, Default LoRA Confinement: ${confinementName}`; } - description = `${isStarred ? 'Starred: ' : ''}${escapeHtml(display)}
${getLine("Title", model.data.title)}${getOptLine("Author", model.data.author)}${getLine("Type", model.data.class)}${interject}${getOptLine('Trigger Phrase', model.data.trigger_phrase)}${getOptLine('Usage Hint', model.data.usage_hint)}${getLine("Description", model.data.description)}
`; + // Add linked preset info if available + let presetLine = ''; + let linkedPreset = getItemPresetLink(this.subType, model.data.name); + if (linkedPreset) { + presetLine = `${getLine("Preset", linkedPreset)}`; + searchableAdded += `, Preset: ${linkedPreset}`; + } + description = `${isStarred ? 'Starred: ' : ''}${escapeHtml(display)}
${getLine("Title", model.data.title)}${presetLine}${getOptLine("Author", model.data.author)}${getLine("Type", model.data.class)}${interject}${getOptLine('Trigger Phrase', model.data.trigger_phrase)}${getOptLine('Usage Hint', model.data.usage_hint)}${getLine("Description", model.data.description)}
`; let cleanForDetails = (val) => val == null ? '(Unset)' : safeHtmlOnly(val).replaceAll('
', ' '); detail_list.push(cleanForDetails(model.data.title), cleanForDetails(model.data.class), cleanForDetails(model.data.usage_hint ?? model.data.trigger_phrase), cleanForDetails(model.data.description)); if (model.data.local && permissions.hasPermission('edit_model_metadata')) { @@ -873,6 +888,7 @@ function directSetModel(model) { if (!model) { return; } + let modelName = null; if (model.name) { let clean = cleanModelName(model.name); forceSetDropdownValue('input_model', clean); @@ -883,6 +899,7 @@ function directSetModel(model) { curModelArch = model.architecture; curModelCompatClass = model.compat_class; curModelSpecialFormat = model.special_format; + modelName = clean; } else if (model.includes(',')) { let [name, width, height, arch, compatClass, specialFormat] = model.split(','); @@ -894,8 +911,53 @@ function directSetModel(model) { curModelArch = arch; curModelCompatClass = compatClass; curModelSpecialFormat = specialFormat; + modelName = name; } reviseBackendFeatureSet(); + + // Look up preset link using the full model name (getItemPresetLink handles extraction) + let presetTitle = getItemPresetLink(model.subType, model.name || modelName); + try { + let autoApply = getUserSetting('ui.autoapplymodelpresets', false); + if (presetTitle) { + currentPresets = removeOtherModelPresets(model.name || modelName); + let presetData = allPresetsUnsorted?.find(p => p.title == presetTitle); + if (presetData) { + if (autoApply) { + applyOnePreset(presetData, false); // false = not nested, allow LoRA preset triggering + // After auto-applying, remove the preset from currentPresets so user can edit those params + currentPresets = currentPresets.filter(p => p.title != presetData.title); + } + else { + // Select the preset if it's not already selected + if (!currentPresets.some(p => p.title == presetData.title)) { + // Add it directly instead of using selectPreset (which toggles) + currentPresets.push(presetData); + } + // Still need to handle LoRA presets for selection (not auto-apply, just selection) + if (presetData.param_map.loras && presetData.param_map.loraweights) { + // Ensure item preset links are loaded before trying to find LoRA preset links + genericRequest('GetMyUserData', {}, data => { + if (data.item_preset_links && itemPresetLinkManager) { + itemPresetLinkManager.loadFromServer(data.item_preset_links); + } + selectOrApplyLoraPresetsFromList(presetData.param_map.loras, presetData.param_map.loraweights); + }); + } + } + updatePresetList(); + presetBrowser?.rerender(); + } + } else { + // No preset linked to this model, but we may have removed other presets above + updatePresetList(); + presetBrowser?.rerender(); + } + } catch (e) { + // Setting may not exist yet, continue without auto-apply + console.warn('[Model Preset] Auto-apply setting error:', e.message); + } + getRequiredElementById('input_model').dispatchEvent(new Event('change')); let aspect = document.getElementById('input_aspectratio'); if (aspect) { @@ -1024,4 +1086,121 @@ function doModelInstallRequiredCheck() { return false; } +/** + * Removes preset links from other models when a model with its own preset is selected. + * This enforces mutual exclusivity of model presets - only one model's preset should be active. + * @param {string} currentModelName - The full path/name of the currently-selected model + * @returns {array} Updated currentPresets array with other models' presets removed + */ +function removeOtherModelPresets(currentModelName) { + if (!sdModelBrowser?.models || !currentModelName) { + return currentPresets; + } + let otherModelPresets = new Set(); + for (let otherModelName in sdModelBrowser.models) { + let otherPresetTitle = getItemPresetLink('Stable-Diffusion', otherModelName); + // Skip presets from the current model and models without presets + if (otherPresetTitle && extractItemKeyNameFromPath(otherModelName) !== extractItemKeyNameFromPath(currentModelName)) { + otherModelPresets.add(otherPresetTitle); + } + } + // Return filtered presets if there are any to remove, otherwise return unchanged + return otherModelPresets.size > 0 + ? currentPresets.filter(p => !otherModelPresets.has(p.title)) + : currentPresets; +} + +/** + * Extract the base filename from a full model/LoRA path. + * This is used as the key for preset linking to ensure links don't break if the user reorganizes + * their files. + * + * Examples: + * - 'SDXL/models/my_model.safetensors' -> 'my_model' + * - 'loras/my_lora.safetensors' -> 'my_lora' + * - 'my_model' -> 'my_model' + * + * @param {string} itemName - The full model/LoRA name (path or filename) + * @returns {string} The base filename without path or extension + */ +function extractItemKeyNameFromPath(itemName) { + if (!itemName) { + return ''; + } + // Remove directory path (handles both / and \ separators) + let lastSlash = Math.max(itemName.lastIndexOf('/'), itemName.lastIndexOf('\\')); + let name = lastSlash !== -1 ? itemName.substring(lastSlash + 1) : itemName; + // Remove .safetensors extension + if (name.endsWith('.safetensors')) { + name = name.substring(0, name.length - '.safetensors'.length); + } + return name; +} + +/** + * Get a preset link for a model or LoRA using the base filename as the key. + * + * @param {string} itemType - Either 'Stable-Diffusion' or 'LoRA' + * @param {string} itemName - The full name (path or filename) + * @returns {string|null} The linked preset title, or null if no link exists + */ +function getItemPresetLink(itemType, itemName) { + if (!itemName) { + return null; + } + let keyName = extractItemKeyNameFromPath(itemName); + return itemPresetLinkManager.getLink(itemType, keyName); +} + +/** + * Save or clear a preset link for a model or LoRA using the base filename as the key. + * This centralizes all preset link saving logic to ensure consistency. + * + * @param {string} itemType - Either 'Stable-Diffusion' or 'LoRA' + * @param {string} itemName - The full name (path or filename) + * @param {string} presetTitle - The preset to link (or empty string to clear) + */ +function saveItemPresetLink(itemType, itemName, presetTitle) { + if (!itemName) { + return; + } + let keyName = extractItemKeyNameFromPath(itemName); + if (presetTitle && presetTitle.trim()) { + genericRequest('SaveItemPresetLink', { 'itemType': itemType, 'itemName': keyName, 'presetTitle': presetTitle }, data => { + itemPresetLinkManager.setLink(itemType, keyName, presetTitle); + }); + } else { + genericRequest('ClearItemPresetLink', { 'itemType': itemType, 'itemName': keyName }, data => { + itemPresetLinkManager.clearLink(itemType, keyName); + }); + } +} + +function applyPresetOverride() { + // Set the dropdown to the first currently-selected preset + if (currentPresets?.length) { + let presetSelect = document.getElementById('edit_model_override_preset_id'); + if (presetSelect) { + presetSelect.value = currentPresets[0].title; + presetSelect.dispatchEvent(new Event('change')); + } + } + updatePresetLinkButtonState('edit_model_override_preset_id'); +} + +function clearPresetOverride() { + let presetSelect = document.getElementById('edit_model_override_preset_id'); + if (presetSelect) { + presetSelect.value = ''; + presetSelect.dispatchEvent(new Event('change')); + } + updatePresetLinkButtonState('edit_model_override_preset_id'); +} + + +function buildModelOverridePresetInput() { + // Pass the full model name; buildPresetLinkSelectorForItem will handle key extraction + buildPresetLinkSelectorForItem('model', curModelMenuModel.name, 'edit_model_override_preset', 'edit_model_override_preset_id'); +} + getRequiredElementById('current_model').addEventListener('change', currentModelChanged); diff --git a/src/wwwroot/js/genpage/gentab/presets.js b/src/wwwroot/js/genpage/gentab/presets.js index 3c65c0adf..b9d59b333 100644 --- a/src/wwwroot/js/genpage/gentab/presets.js +++ b/src/wwwroot/js/genpage/gentab/presets.js @@ -10,6 +10,45 @@ class PresetHelpers { } } +/** Manages preset links for models and LoRAs. */ +class ItemPresetLinkManager { + constructor() { + this.links = {}; // { 'model:name': 'preset_title', 'lora:name': 'preset_title', ... } + } + + getLink(itemType, itemName) { + let key = `${itemType}:${itemName}`; + return this.links[key] || null; + } + + setLink(itemType, itemName, presetTitle) { + let key = `${itemType}:${itemName}`; + if (presetTitle && presetTitle.trim()) { + this.links[key] = presetTitle; + } else { + delete this.links[key]; + } + } + + clearLink(itemType, itemName) { + let key = `${itemType}:${itemName}`; + delete this.links[key]; + } + + hasLink(itemType, itemName) { + return this.getLink(itemType, itemName) !== null; + } + + loadFromServer(data) { + if (data && typeof data === 'object') { + this.links = { ...data }; + } + } +} + +/** Instance of ItemPresetLinkManager for managing model and LoRA preset links */ +let itemPresetLinkManager = new ItemPresetLinkManager(); + /** Collection of helper functions and data related to presets, just an instance of {@link PresetHelpers}. */ let presetHelpers = new PresetHelpers(); @@ -250,7 +289,17 @@ function selectInitialPresetList() { } } -function applyOnePreset(preset) { +function applyOnePreset(preset, isNestedPreset = false) { + // Collect LoRA updates before applying to handle merging + let presetLoras = preset.param_map.loras ? preset.param_map.loras.split(',').map(l => l.trim()).filter(l => l.length > 0) : []; + let presetWeights = preset.param_map.loraweights ? preset.param_map.loraweights.split(',').map(w => { + let parsed = parseFloat(w.trim()); + return isNaN(parsed) ? 0 : parsed; + }) : []; + + // Track if we've handled loraweights through loras processing + let loraweightsHandled = false; + for (let key of Object.keys(preset.param_map)) { let param = gen_param_types.filter(p => p.id == key)[0]; if (param) { @@ -267,11 +316,47 @@ function applyOnePreset(preset) { val = val.replace("{value}", elem.value); } } - else if (key == 'loras' && rawVal) { - val = rawVal + "," + val; + else if (key == 'loras') { + if (rawVal) { + // Merge LoRA lists: update weights for existing LoRAs, add new ones + let currentLoras = rawVal.split(',').map(l => l.trim()).filter(l => l.length > 0); + let weightsParam = gen_param_types.find(p => p.id == 'loraweights'); + if (!weightsParam) { + console.warn('[Preset] Warning: loraweights parameter not found when merging LoRA lists'); + } + let weightsVal = weightsParam ? getInputVal(weightsParam) : ''; + let currentWeights = (weightsVal || '').split(',').map(w => { + let parsed = parseFloat(w.trim()); + return isNaN(parsed) ? 0 : parsed; + }); + // Update weights for LoRAs that are already present + for (let i = 0; i < presetLoras.length; i++) { + let idx = currentLoras.indexOf(presetLoras[i]); + if (idx >= 0) { + // LoRA already exists, update its weight + currentWeights[idx] = presetWeights[i] !== undefined ? presetWeights[i] : 0; + } else { + // New LoRA, add it + currentLoras.push(presetLoras[i]); + currentWeights.push(presetWeights[i] !== undefined ? presetWeights[i] : 0); + } + } + val = currentLoras.join(','); + // Also update the weights value + if (weightsParam) { + setDirectParamValue(weightsParam, currentWeights.join(',')); + } + loraweightsHandled = true; + } else { + // No existing LoRAs, just use the preset's LoRAs + val = preset.param_map.loras; + } } - else if (key == 'loraweights' && rawVal) { - val = rawVal + "," + val; + else if (key == 'loraweights') { + // Skip - weights are handled in loras section + if (loraweightsHandled) { + continue; + } } setDirectParamValue(param, val); if (param.group && param.group.toggles) { @@ -280,6 +365,110 @@ function applyOnePreset(preset) { doToggleGroup(`input_group_content_${param.group.id}`); } } + } + // Handle LoRA preset selection/application (only if not already a nested preset to prevent recursion) + if (!isNestedPreset && preset.param_map.loras) { + selectOrApplyLoraPresetsFromList(preset.param_map.loras, preset.param_map.loraweights); + } +} + +/** + * Common helper to apply or select a preset with proper UI updates. + * @param {string} source - Human-readable source for logging (e.g., 'LoRA', 'Model') + * @param {string} itemName - Name of the item triggering the preset + * @param {object} presetData - The preset data object + * @param {boolean} isNested - Whether this is a nested call (prevent recursion) + * @param {boolean} shouldUpdateUI - Whether to update UI after selection + */ +function applyOrSelectPreset(source, itemName, presetData, isNested, shouldUpdateUI = false) { + let autoApply = getUserSetting('ui.autoapplymodelpresets', false); + if (autoApply) { + console.info(`[${source} Preset] Auto-applying preset ${presetData.title} for ${itemName}`); + applyOnePreset(presetData, isNested); + } else if (!currentPresets.some(p => p.title == presetData.title)) { + // Select the preset if it's not already selected + currentPresets.push(presetData); + if (shouldUpdateUI) { + updatePresetList(); + presetBrowser?.rerender(); + } + } +} + +/** + * Handle LoRA preset selection/application for LoRAs loaded by a preset. + * Parses the loras and loraweights strings to find non-zero weight LoRAs and select/apply their presets. + * @param {string} lorasStr - Comma-separated list of LoRA names + * @param {string} weightsStr - Comma-separated list of LoRA weights + */ +function selectOrApplyLoraPresetsFromList(lorasStr, weightsStr) { + if (!lorasStr) { + return; + } + if (!weightsStr) { + return; + } + try { + let loras = lorasStr.split(',').map(l => l.trim()).filter(l => l.length > 0); + let weights = weightsStr.split(',').map(w => { + let parsed = parseFloat(w.trim()); + return isNaN(parsed) ? 0 : parsed; + }); + let hasAnyPresets = false; + let autoApply = getUserSetting('ui.autoapplymodelpresets', false); + let ignoreZeroWeightLoraPresets = getUserSetting('ui.ignoreModelPresetZeroWeightLoraPresets', false); + for (let i = 0; i < loras.length; i++) { + let loraName = loras[i]; + let weight = weights[i] !== undefined ? weights[i] : 1; + // Only process LoRAs with non-zero weight (if setting is enabled) + if (ignoreZeroWeightLoraPresets && weight == 0) { + continue; + } + // Use the centralized function to get preset links (handles key extraction) + let presetTitle = getItemPresetLink('lora', loraName); + if (!presetTitle) { + continue; + } + let presetData = allPresetsUnsorted?.find(p => p.title == presetTitle); + if (!presetData) { + continue; + } + hasAnyPresets = true; + if (autoApply) { + // Directly apply the LoRA preset (pass true for isNested to prevent recursion) + applyOnePreset(presetData, true); + } else { + // In non-auto-apply mode, just select the preset + if (!currentPresets.some(p => p.title == presetData.title)) { + currentPresets.push(presetData); + } + } + } + // Update UI once if we processed any presets + if (hasAnyPresets) { + updatePresetList(); + presetBrowser?.rerender(); + } + } catch (e) { + console.warn('[LoRA Preset] Error handling LoRA presets:', e.message); + } +} + +/** + * Handle LoRA preset selection/application when a LoRA is manually selected. + * @param {string} loraName - The name of the LoRA being selected + */ +function selectOrApplyLoraPresetOnSelection(loraName) { + try { + let presetTitle = getItemPresetLink('lora', loraName); + if (presetTitle) { + let presetData = allPresetsUnsorted?.find(p => p.title == presetTitle); + if (presetData) { + applyOrSelectPreset('LoRA', loraName, presetData, true, true); + } + } + } catch (e) { + console.warn('[LoRA Preset] Error handling LoRA preset on selection:', e.message); } } @@ -410,6 +599,10 @@ function listPresetFolderAndFiles(path, isRefresh, callback, depth) { if (isRefresh) { genericRequest('GetMyUserData', {}, data => { allPresetsUnsorted = data.presets; + // Load user's item preset links from server into the manager + if (data.item_preset_links && itemPresetLinkManager) { + itemPresetLinkManager.loadFromServer(data.item_preset_links); + } proc(); }); } @@ -434,7 +627,7 @@ function describePreset(preset) { } } ]; let paramText = Object.keys(preset.data.param_map).map(key => `${key}: ${preset.data.param_map[key]}`); - let description = `${preset.data.title}:\n${preset.data.description}\n\n${paramText.join('\n')}`; + let description = `${preset.data.title}:\n${preset.data.description ? preset.data.description + '\n' : ''}\n${paramText.join('\n')}`; let className = currentPresets.some(p => p.title == preset.data.title) ? 'preset-block-selected preset-block' : 'preset-block'; let name = preset.data.title; let index = name.lastIndexOf('/'); @@ -784,3 +977,83 @@ function closeExportPresetViewer() { function closeImportPresetViewer() { $('#import_presets_modal').modal('hide'); } + +/** + * Generic function to build a preset link selector for both models and LoRAs. + * @param {string} itemType - 'model' or 'lora' + * @param {string} itemName - The name of the model or LoRA + * @param {string} containerId - The ID of the container element to populate + * @param {string} selectId - The ID to assign to the select element (e.g., 'edit_model_override_preset_id') + */ +function buildPresetLinkSelectorForItem(itemType, itemName, containerId, selectId) { + let container = getRequiredElementById(containerId); + container.innerHTML = ''; + + // Build list of all presets + let compatiblePresets = []; + if (allPresetsUnsorted && allPresetsUnsorted.length > 0) { + for (let preset of allPresetsUnsorted) { + let presetData = preset.data || preset; + if (presetData && presetData.title) { + compatiblePresets.push(presetData); + } + } + } + + // Create select element that UIImprovementHandler will convert to a popover + let select = document.createElement('select'); + select.id = selectId; + select.className = 'modal_text_extra'; + + let option = document.createElement('option'); + option.value = ''; + option.innerText = '-- Select a Preset --'; + select.appendChild(option); + + for (let preset of compatiblePresets) { + option = document.createElement('option'); + option.value = preset.title; + option.innerText = preset.title; + select.appendChild(option); + } + + // Set current value from itemPresetLinkManager using centralized key extraction + let currentLink = getItemPresetLink(itemType, itemName); + if (currentLink) { + select.value = currentLink; + } + + select.addEventListener('change', () => { + // Update button state when selection changes + updatePresetLinkButtonState(selectId); + }); + select.addEventListener('input', () => { + updatePresetLinkButtonState(selectId); + }); + + container.appendChild(select); + updatePresetLinkButtonState(selectId); +} + +/** + * Update the enabled state of Set/Clear buttons based on preset selector state. + * @param {string} selectId - The ID of the select element + */ +function updatePresetLinkButtonState(selectId) { + let select = document.getElementById(selectId); + if (!select) return; + + // Determine button IDs based on select element ID + let baseId = select.id.replace('_id', ''); + let clearBtn = document.getElementById(`${baseId}_clear_btn`); + let setBtn = document.getElementById(`${baseId}_set_btn`); + + if (clearBtn) { + clearBtn.classList.toggle('disabled-dim', select.value == ''); + } + + if (setBtn) { + // Dim Set button if no presets are currently selected + setBtn.classList.toggle('disabled-dim', !currentPresets?.length); + } +} diff --git a/src/wwwroot/js/genpage/main.js b/src/wwwroot/js/genpage/main.js index c0eb5c4ca..051487a1a 100644 --- a/src/wwwroot/js/genpage/main.js +++ b/src/wwwroot/js/genpage/main.js @@ -373,6 +373,10 @@ function loadUserData(callback) { language = data.language; } allPresetsUnsorted = data.presets; + // Load user's item preset links from server into the manager + if (data.item_preset_links && itemPresetLinkManager) { + itemPresetLinkManager.loadFromServer(data.item_preset_links); + } sortPresets(); presetBrowser.lightRefresh(); if (shouldApplyDefault) {