From dcc180bcfc0133588e6cbf39302835332cd4fb5b Mon Sep 17 00:00:00 2001
From: BetaDoggo <alexnunet@gmail.com>
Date: Sat, 3 Aug 2024 17:21:21 -0400
Subject: [PATCH] small over step fix

---
 nodes.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/nodes.py b/nodes.py
index cbb3c9c..1715465 100644
--- a/nodes.py
+++ b/nodes.py
@@ -30,6 +30,9 @@ class FalFluxAPI:
     CATEGORY = "FalAPI"
 
     def generate_image(self, prompt, endpoint, width, height, steps, api_key, seed, cfg_dev_and_pro):
+        #prevent too many steps error
+        if endpoint == "schnell (4+ steps)" and steps > 8:
+            steps = 8
         #set endpoint
         models = {
             "schnell (4+ steps)": "fal-ai/flux/schnell",