Download javascript/edit-attention.js from nolanaatama/stable-diffusion-webui: direct link, hf CLI and curl.
- Browser
- Download file 3.01 kB
-
https://huggingface.co/nolanaatama/stable-diffusion-webui/resolve/main/javascript/edit-attention.js
- Command line
-
hf download hf://nolanaatama/stable-diffusion-webui/javascript/edit-attention.js
-
curl -L -o edit-attention.js https://huggingface.co/nolanaatama/stable-diffusion-webui/resolve/main/javascript/edit-attention.js
3.01 kB
| addEventListener('keydown', (event) => { | |
| let target = event.originalTarget || event.composedPath()[0]; | |
| if (!target.matches("#toprow textarea.gr-text-input[placeholder]")) return; | |
| if (! (event.metaKey || event.ctrlKey)) return; | |
| let plus = "ArrowUp" | |
| let minus = "ArrowDown" | |
| if (event.key != plus && event.key != minus) return; | |
| let selectionStart = target.selectionStart; | |
| let selectionEnd = target.selectionEnd; | |
| // If the user hasn't selected anything, let's select their current parenthesis block | |
| if (selectionStart === selectionEnd) { | |
| // Find opening parenthesis around current cursor | |
| const before = target.value.substring(0, selectionStart); | |
| let beforeParen = before.lastIndexOf("("); | |
| if (beforeParen == -1) return; | |
| let beforeParenClose = before.lastIndexOf(")"); | |
| while (beforeParenClose !== -1 && beforeParenClose > beforeParen) { | |
| beforeParen = before.lastIndexOf("(", beforeParen - 1); | |
| beforeParenClose = before.lastIndexOf(")", beforeParenClose - 1); | |
| } | |
| // Find closing parenthesis around current cursor | |
| const after = target.value.substring(selectionStart); | |
| let afterParen = after.indexOf(")"); | |
| if (afterParen == -1) return; | |
| let afterParenOpen = after.indexOf("("); | |
| while (afterParenOpen !== -1 && afterParen > afterParenOpen) { | |
| afterParen = after.indexOf(")", afterParen + 1); | |
| afterParenOpen = after.indexOf("(", afterParenOpen + 1); | |
| } | |
| if (beforeParen === -1 || afterParen === -1) return; | |
| // Set the selection to the text between the parenthesis | |
| const parenContent = target.value.substring(beforeParen + 1, selectionStart + afterParen); | |
| const lastColon = parenContent.lastIndexOf(":"); | |
| selectionStart = beforeParen + 1; | |
| selectionEnd = selectionStart + lastColon; | |
| target.setSelectionRange(selectionStart, selectionEnd); | |
| } | |
| event.preventDefault(); | |
| if (selectionStart == 0 || target.value[selectionStart - 1] != "(") { | |
| target.value = target.value.slice(0, selectionStart) + | |
| "(" + target.value.slice(selectionStart, selectionEnd) + ":1.0)" + | |
| target.value.slice(selectionEnd); | |
| target.focus(); | |
| target.selectionStart = selectionStart + 1; | |
| target.selectionEnd = selectionEnd + 1; | |
| } else { | |
| end = target.value.slice(selectionEnd + 1).indexOf(")") + 1; | |
| weight = parseFloat(target.value.slice(selectionEnd + 1, selectionEnd + 1 + end)); | |
| if (isNaN(weight)) return; | |
| if (event.key == minus) weight -= 0.1; | |
| if (event.key == plus) weight += 0.1; | |
| weight = parseFloat(weight.toPrecision(12)); | |
| target.value = target.value.slice(0, selectionEnd + 1) + | |
| weight + | |
| target.value.slice(selectionEnd + 1 + end - 1); | |
| target.focus(); | |
| target.selectionStart = selectionStart; | |
| target.selectionEnd = selectionEnd; | |
| } | |
| // Since we've modified a Gradio Textbox component manually, we need to simulate an `input` DOM event to ensure its | |
| // internal Svelte data binding remains in sync. | |
| target.dispatchEvent(new Event("input", { bubbles: true })); | |
| }); | |