make disabling safety checker explicit

2024-08-11 22:45:30 -05:00
parent b7bd29ffa0
commit 05608b2acb
1 changed files with 2 additions and 0 deletions
@@ -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()