Allow configuring FAL safety settings

elizaOS · Dec 3, 2024 · eed8b06 · eed8b06
1 parent a0992ff
commit eed8b06
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/packages/core/src/generation.ts b/packages/core/src/generation.ts
@@ -907,7 +907,7 @@ export const generateImage = async (
                 const blob = await imageResponse.blob();
                 const arrayBuffer = await blob.arrayBuffer();
                 const base64 = Buffer.from(arrayBuffer).toString('base64');
-                
+
                 // Return with proper MIME type
                 return `data:image/jpeg;base64,${base64}`;
             }));
@@ -930,7 +930,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")