Skip to content

Commit eed8b06

Browse files
committedDec 3, 2024
Allow configuring FAL safety settings
1 parent a0992ff commit eed8b06

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed
 

‎packages/core/src/generation.ts

+3-2
Original file line numberDiff line numberDiff line change
@@ -907,7 +907,7 @@ export const generateImage = async (
907907
const blob = await imageResponse.blob();
908908
const arrayBuffer = await blob.arrayBuffer();
909909
const base64 = Buffer.from(arrayBuffer).toString('base64');
910-
910+
911911
// Return with proper MIME type
912912
return `data:image/jpeg;base64,${base64}`;
913913
}));
@@ -930,7 +930,8 @@ export const generateImage = async (
930930
num_inference_steps: modelSettings?.steps ?? 50,
931931
guidance_scale: data.guidanceScale || 3.5,
932932
num_images: data.count,
933-
enable_safety_checker: true,
933+
enable_safety_checker: runtime.getSetting("FAL_AI_ENABLE_SAFETY_CHECKER") === "true",
934+
safety_tolerance: Number(runtime.getSetting("FAL_AI_SAFETY_TOLERANCE") || "2"),
934935
output_format: "png" as const,
935936
seed: data.seed ?? 6252023,
936937
...(runtime.getSetting("FAL_AI_LORA_PATH")

0 commit comments

Comments
 (0)
Please sign in to comment.