Skip to content

Commit dfab4c1

Browse files
authored
Merge pull request #1293 from ai16z/patch/handle_provider
fix: add missing claude vertex case to handleProvider
2 parents 907c5a5 + 0eb98a3 commit dfab4c1

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

packages/core/src/generation.ts

+7-2
Original file line numberDiff line numberDiff line change
@@ -1066,8 +1066,12 @@ export const generateImage = async (
10661066
num_inference_steps: modelSettings?.steps ?? 50,
10671067
guidance_scale: data.guidanceScale || 3.5,
10681068
num_images: data.count,
1069-
enable_safety_checker: runtime.getSetting("FAL_AI_ENABLE_SAFETY_CHECKER") === "true",
1070-
safety_tolerance: Number(runtime.getSetting("FAL_AI_SAFETY_TOLERANCE") || "2"),
1069+
enable_safety_checker:
1070+
runtime.getSetting("FAL_AI_ENABLE_SAFETY_CHECKER") ===
1071+
"true",
1072+
safety_tolerance: Number(
1073+
runtime.getSetting("FAL_AI_SAFETY_TOLERANCE") || "2"
1074+
),
10711075
output_format: "png" as const,
10721076
seed: data.seed ?? 6252023,
10731077
...(runtime.getSetting("FAL_AI_LORA_PATH")
@@ -1368,6 +1372,7 @@ export async function handleProvider(
13681372
case ModelProviderName.AKASH_CHAT_API:
13691373
return await handleOpenAI(options);
13701374
case ModelProviderName.ANTHROPIC:
1375+
case ModelProviderName.CLAUDE_VERTEX:
13711376
return await handleAnthropic(options);
13721377
case ModelProviderName.GROK:
13731378
return await handleGrok(options);

0 commit comments

Comments
 (0)