Adds support for Knob display option in IntWidget

Comfy-Org · Feb 25, 2025 · faa8923 · faa8923
1 parent 79be18a
commit faa8923
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/src/composables/widgets/useIntWidget.ts b/src/composables/widgets/useIntWidget.ts
@@ -20,12 +20,18 @@ export const useIntWidget = () => {
   ) => {
     const settingStore = useSettingStore()
     const sliderEnabled = !settingStore.get('Comfy.DisableSliders')
+    const knobEnabled = !settingStore.get('Comfy.DisableKnobs')
     const inputOptions = inputData[1]
-    const widgetType = sliderEnabled
-      ? inputOptions?.display === 'slider'
-        ? 'slider'
-        : 'number'
-      : 'number'
+    const widgetType = (() => {
+      switch (inputOptions?.display) {
+        case 'slider':
+          return sliderEnabled ? 'slider' : 'number'
+        case 'knob':
+          return knobEnabled ? 'knob' : 'number'
+        default:
+          return 'number'
+      }
+    })()
 
     const { val, config } = getNumberDefaults(inputOptions, {
       defaultStep: 1,