small over step fix

2024-08-03 17:21:21 -04:00
parent aacdd16d1d
commit dcc180bcfc
1 changed files with 3 additions and 0 deletions
@@ -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",