diff --git a/packages/core/src/generation.ts b/packages/core/src/generation.ts index ee0bea2e1b..59b620d648 100644 --- a/packages/core/src/generation.ts +++ b/packages/core/src/generation.ts @@ -1065,7 +1065,8 @@ export const generateImage = async ( num_inference_steps: modelSettings?.steps ?? 50, guidance_scale: data.guidanceScale || 3.5, num_images: data.count, - enable_safety_checker: true, + enable_safety_checker: runtime.getSetting("FAL_AI_ENABLE_SAFETY_CHECKER") === "true", + safety_tolerance: Number(runtime.getSetting("FAL_AI_SAFETY_TOLERANCE") || "2"), output_format: "png" as const, seed: data.seed ?? 6252023, ...(runtime.getSetting("FAL_AI_LORA_PATH")