From 05608b2acbe4e18910018ff478f56591b8f7bb5f Mon Sep 17 00:00:00 2001
From: Alex Nunet <alexnunet@gmail.com>
Date: Sun, 11 Aug 2024 22:45:30 -0500
Subject: [PATCH] make disabling safety checker explicit

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

diff --git a/nodes.py b/nodes.py
index edb4560..fdd0762 100644
--- a/nodes.py
+++ b/nodes.py
@@ -280,6 +280,7 @@ class FalFluxI2IAPI:
                 "height": height},
             "strength": strength,
             "guidance_scale": cfg,
+            "enable_safety_checker": False,
             "num_inference_steps": steps,
             "num_images": 1, #Hardcoded to 1 for now
         })
@@ -372,6 +373,7 @@ class FalFluxAPI:
                 "height": height,
             },
             "num_inference_steps": steps,
+            "enable_safety_checker": False,
             "num_images": 1,}  #Hardcoded to 1 for now
         )
         result = handler.get()