wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-29 15:50:22 +08:00
Code Issues Actions 22 Packages Projects Releases Wiki Activity

Merge branch 'comfyanonymous:master' into feature/blockweights

Browse Source
This commit is contained in:
ltdrdata 2023-04-18 20:45:50 +09:00 committed by GitHub
commit 0f956d5535
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
web/extensions/core/editAttention.js
View File

@ -2,10 +2,21 @@ import { app } from "/scripts/app.js";
// Allows you to edit the attention weight by holding ctrl (or cmd) and using the up/down arrow keys // Allows you to edit the attention weight by holding ctrl (or cmd) and using the up/down arrow keys
const id = "Comfy.EditAttention";
app.registerExtension({ app.registerExtension({
name:id, name: "Comfy.EditAttention",
init() { init() {
const editAttentionDelta = app.ui.settings.addSetting({
id: "Comfy.EditAttention.Delta",
name: "Ctrl+up/down precision",
type: "slider",
attrs: {
min: 0.01,
max: 0.5,
step: 0.01,
},
defaultValue: 0.05,
});
function incrementWeight(weight, delta) { function incrementWeight(weight, delta) {
const floatWeight = parseFloat(weight); const floatWeight = parseFloat(weight);
if (isNaN(floatWeight)) return weight; if (isNaN(floatWeight)) return weight;
@ -58,7 +69,7 @@ name:id,
function editAttention(event) { function editAttention(event) {
const inputField = event.composedPath()[0]; const inputField = event.composedPath()[0];
const delta = 0.025; const delta = parseFloat(editAttentionDelta.value);
if (inputField.tagName !== "TEXTAREA") return; if (inputField.tagName !== "TEXTAREA") return;
if (!(event.key === "ArrowUp" || event.key === "ArrowDown")) return; if (!(event.key === "ArrowUp" || event.key === "ArrowDown")) return;
@ -125,7 +136,7 @@ name:id,
// Increment the weight // Increment the weight
const weightDelta = event.key === "ArrowUp" ? delta : -delta; const weightDelta = event.key === "ArrowUp" ? delta : -delta;
const updatedText = selectedText.replace(/(.*:)(\d+(\.\d+)?)(.*)/, (match, prefix, weight, _, suffix) => { const updatedText = selectedText.replace(/(.*:)(\d+(\.\d+)?)(.*)/, (match, prefix, weight, _, suffix) => {
return prefix + incrementWeight(weight, weightDelta) + suffix; return prefix + incrementWeight(weight, weightDelta) + suffix;
}); });
inputField.setRangeText(updatedText, start, end, "select"); inputField.setRangeText(updatedText, start, end, "select");