wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-10 13:32:36 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

Add toggle for float rounding

Browse Source
This commit is contained in:
City 2023-09-17 16:05:56 +02:00
parent 321c5fa295
commit 80dfeb089b
2 changed files with 28 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
web/scripts/ui.js
View File

@ -577,6 +577,13 @@ export class ComfyUI {
defaultValue: false, defaultValue: false,
}); });
this.settings.addSetting({
id: "Comfy.DisableFloatRounding",
name: "Disable rounding floats (requires page reload).",
type: "boolean",
defaultValue: false,
});
const fileInput = $el("input", { const fileInput = $el("input", {
id: "comfy-file-input", id: "comfy-file-input",
type: "file", type: "file",

25
web/scripts/widgets.js
View File

@ -1,6 +1,6 @@
import { api } from "./api.js" import { api } from "./api.js"
function getNumberDefaults(inputData, defaultStep) { function getNumberDefaults(inputData, defaultStep, app) {
let defaultVal = inputData[1]["default"]; let defaultVal = inputData[1]["default"];
let { min, max, step } = inputData[1]; let { min, max, step } = inputData[1];
@ -8,11 +8,16 @@ function getNumberDefaults(inputData, defaultStep) {
if (min == undefined) min = 0; if (min == undefined) min = 0;
if (max == undefined) max = 2048; if (max == undefined) max = 2048;
if (step == undefined) step = defaultStep; if (step == undefined) step = defaultStep;
// precision is the number of decimal places to show. // precision is the number of decimal places to show.
// by default, display the the smallest number of decimal places such that changes of size step are visible. // when using the old behavior, it defaults to 3 decimal places
let precision = Math.max(-Math.floor(Math.log10(step)),0) let precision = 3;
// by default, round the value to those decimal places shown. let round = 0.001;
let round = Math.round(1000000*Math.pow(0.1,precision))/1000000; if (!app.ui.settings.getSettingValue("Comfy.DisableFloatRounding")) {
// display the the smallest number of decimal places such that changes of size step are visible.
precision = Math.max(-Math.floor(Math.log10(step)),0);
// round the value to those decimal places shown.
round = Math.round(1000000*Math.pow(0.1,precision))/1000000;
}
return { val: defaultVal, config: { min, max, step: 10.0 * step, round, precision } }; return { val: defaultVal, config: { min, max, step: 10.0 * step, round, precision } };
} }
@ -268,15 +273,19 @@ export const ComfyWidgets = {
"INT:noise_seed": seedWidget, "INT:noise_seed": seedWidget,
FLOAT(node, inputName, inputData, app) { FLOAT(node, inputName, inputData, app) {
let widgetType = isSlider(inputData[1]["display"], app); let widgetType = isSlider(inputData[1]["display"], app);
const { val, config } = getNumberDefaults(inputData, 0.5); const { val, config } = getNumberDefaults(inputData, 0.5, app);
if (!app.ui.settings.getSettingValue("Comfy.DisableFloatRounding")) {
return { widget: node.addWidget(widgetType, inputName, val, return { widget: node.addWidget(widgetType, inputName, val,
function (v) { function (v) {
this.value = Math.round(v/config.round)*config.round; this.value = Math.round(v/config.round)*config.round;
}, config) }; }, config) };
} else {
return { widget: node.addWidget(widgetType, inputName, val, () => {}, config) };
}
}, },
INT(node, inputName, inputData, app) { INT(node, inputName, inputData, app) {
let widgetType = isSlider(inputData[1]["display"], app); let widgetType = isSlider(inputData[1]["display"], app);
const { val, config } = getNumberDefaults(inputData, 1); const { val, config } = getNumberDefaults(inputData, 1, app);
Object.assign(config, { precision: 0 }); Object.assign(config, { precision: 0 });
return { return {
widget: node.addWidget( widget: node.addWidget(