Merge pull request #61 from Visionatrix/feat/flows/missing-steps_count

Added missing `steps count` param to basic flows
Visionatrix · Nov 15, 2024 · ce12319 · ce12319
2 parents 063a420 + b24a9e5
commit ce12319
Show file tree

Hide file tree

Showing 4 changed files with 110 additions and 14 deletions.
diff --git a/flows/juggernaut_xl.json b/flows/juggernaut_xl.json
@@ -20,7 +20,10 @@
   "27": {
     "inputs": {
       "seed": 376336750880278,
-      "steps": 50,
+      "steps": [
+        "282",
+        0
+      ],
       "cfg": [
         "280",
         0
@@ -188,7 +191,10 @@
   "270": {
     "inputs": {
       "model_type": "SDXL",
-      "steps": 20,
+      "steps": [
+        "283",
+        0
+      ],
       "denoise": 1
     },
     "class_type": "AlignYourStepsScheduler",
@@ -231,7 +237,7 @@
       "documentation": "https://visionatrix.github.io/VixFlowsDocs/Flows/Juggernaut_XL/",
       "license": "",
       "tags": "[\"general\", \"simple\"]",
-      "version": "1.0.1",
+      "version": "1.0.2",
       "requires": "[]",
       "is_seed_supported": true,
       "is_count_supported": true,
@@ -323,5 +329,39 @@
     "_meta": {
       "title": "VixUI-CheckboxLogic"
     }
+  },
+  "282": {
+    "inputs": {
+      "value": 50,
+      "display_name": "Steps number to generate",
+      "optional": true,
+      "advanced": true,
+      "min": 30,
+      "max": 60,
+      "step": 1,
+      "order": 97,
+      "custom_id": "steps_count",
+      "hidden": false
+    },
+    "class_type": "VixUiRangeInt",
+    "_meta": {
+      "title": "VixUI-RangeInt"
+    }
+  },
+  "283": {
+    "inputs": {
+      "python_expression": "a//3",
+      "print_to_console": "False",
+      "a": [
+        "282",
+        0
+      ],
+      "b": 0,
+      "c": 0
+    },
+    "class_type": "Evaluate Integers",
+    "_meta": {
+      "title": "Evaluate Integers"
+    }
   }
 }
diff --git a/flows/mobius_xl.json b/flows/mobius_xl.json
@@ -11,7 +11,10 @@
   "27": {
     "inputs": {
       "seed": 1,
-      "steps": 50,
+      "steps": [
+        "299",
+        0
+      ],
       "cfg": [
         "297",
         0
@@ -190,7 +193,7 @@
       "documentation": "https://visionatrix.github.io/VixFlowsDocs/Flows/Mobius_XL/",
       "license": "",
       "tags": "[\"general\", \"simple\"]",
-      "version": "1.0.1",
+      "version": "1.0.2",
       "requires": "[]",
       "is_seed_supported": true,
       "is_count_supported": true,
@@ -323,5 +326,23 @@
     "_meta": {
       "title": "VixUI-RangeInt"
     }
+  },
+  "299": {
+    "inputs": {
+      "value": 50,
+      "display_name": "Steps number to generate",
+      "optional": true,
+      "advanced": true,
+      "min": 20,
+      "max": 60,
+      "step": 1,
+      "order": 97,
+      "custom_id": "steps_count",
+      "hidden": false
+    },
+    "class_type": "VixUiRangeInt",
+    "_meta": {
+      "title": "VixUI-RangeInt"
+    }
   }
 }
diff --git a/flows/playground_2_5_aesthetic.json b/flows/playground_2_5_aesthetic.json
@@ -71,7 +71,10 @@
   "238": {
     "inputs": {
       "seed": 5,
-      "steps": 50,
+      "steps": [
+        "284",
+        0
+      ],
       "cfg": [
         "256",
         0
@@ -140,7 +143,10 @@
   "247": {
     "inputs": {
       "model_type": "SDXL",
-      "steps": 20,
+      "steps": [
+        "285",
+        0
+      ],
       "denoise": 1
     },
     "class_type": "AlignYourStepsScheduler",
@@ -205,7 +211,7 @@
       "documentation": "https://visionatrix.github.io/VixFlowsDocs/Flows/Playground_2_5_aesthetic/",
       "license": "",
       "tags": "[\"general\", \"simple\"]",
-      "version": "1.1.1",
+      "version": "1.1.2",
       "requires": "[]",
       "is_seed_supported": true,
       "is_count_supported": true,
@@ -331,5 +337,39 @@
     "_meta": {
       "title": "VixUI-RangeInt"
     }
+  },
+  "284": {
+    "inputs": {
+      "value": 50,
+      "display_name": "Steps number to generate",
+      "optional": true,
+      "advanced": true,
+      "min": 30,
+      "max": 60,
+      "step": 1,
+      "order": 97,
+      "custom_id": "steps_count",
+      "hidden": false
+    },
+    "class_type": "VixUiRangeInt",
+    "_meta": {
+      "title": "VixUI-RangeInt"
+    }
+  },
+  "285": {
+    "inputs": {
+      "python_expression": "a//3",
+      "print_to_console": "False",
+      "a": [
+        "284",
+        0
+      ],
+      "b": 0,
+      "c": 0
+    },
+    "class_type": "Evaluate Integers",
+    "_meta": {
+      "title": "Evaluate Integers"
+    }
   }
 }
diff --git a/scripts/benchmarks/benchmark.py b/scripts/benchmarks/benchmark.py
@@ -110,15 +110,13 @@ def validate_flows_test_cases(flows_test_cases: list[FlowTest]):
                 input_params={
                     "prompt": "green apple",
                     "fast_run": False,
-                    "steps_count": 45,
                 },
             ),
             TestCase(
                 name="fast_run",
                 input_params={
                     "prompt": "green apple",
                     "fast_run": True,
-                    "steps_count": 45,
                 },
             ),
         ],
@@ -152,15 +150,13 @@ def validate_flows_test_cases(flows_test_cases: list[FlowTest]):
                 input_params={
                     "prompt": "green apple",
                     "fast_run": True,
-                    "steps_count": 45,
                 },
             ),
             TestCase(
                 name="default",
                 input_params={
                     "prompt": "green apple",
                     "fast_run": False,
-                    "steps_count": 45,
                 },
             ),
         ],
@@ -173,15 +169,13 @@ def validate_flows_test_cases(flows_test_cases: list[FlowTest]):
                 input_params={
                     "prompt": "green apple",
                     "fast_run": False,
-                    "steps_count": 45,
                 },
             ),
             TestCase(
                 name="fast_run",
                 input_params={
                     "prompt": "green apple",
                     "fast_run": True,
-                    "steps_count": 45,
                 },
             ),
         ],
@@ -1098,6 +1092,7 @@ async def select_worker():
                 selected_worker_id = workers[choice]["worker_id"]
                 print(f"Selected worker: '{selected_worker_id}'")
                 SELECTED_WORKER = workers[choice]
+                return
             else:
                 print("Invalid selection. Please try again.")
         except ValueError: