diff --git a/src/server/ai/provider/cloudflare.ts b/src/server/ai/provider/cloudflare.ts index 99a1c48..1a9b1c0 100644 --- a/src/server/ai/provider/cloudflare.ts +++ b/src/server/ai/provider/cloudflare.ts @@ -181,6 +181,15 @@ const Cloudflare: CloudflareProvider = { }, enabledByDefault: true, models: [ + { + id: "@cf/black-forest-labs/flux-2-klein-4b", + name: "FLUX.2 [Klein] - 4B", + ability: "i2i", + maxInputImages: 4, + enabledByDefault: true, + supportedAspectRatios: ["1:1", "16:9", "9:16", "4:3", "3:4"], + inputType: "FormData", + }, { id: "@cf/black-forest-labs/flux-2-dev", name: "FLUX.2-dev", diff --git a/src/server/ai/provider/fal.ts b/src/server/ai/provider/fal.ts index 26daee8..612b3f3 100644 --- a/src/server/ai/provider/fal.ts +++ b/src/server/ai/provider/fal.ts @@ -44,6 +44,20 @@ const Fal: AiProvider = { maxInputImages: 4, enabledByDefault: true, }, + { + id: "fal-ai/flux-2/klein/9b", + name: "FLUX.2 [Klein] - 9B", + ability: "i2i", + maxInputImages: 4, + enabledByDefault: true, + }, + { + id: "fal-ai/flux-2/klein/4b", + name: "FLUX.2 [Klein] - 4B", + ability: "i2i", + maxInputImages: 4, + enabledByDefault: true, + }, { id: "fal-ai/flux-pro/kontext/max", name: "FLUX.1 Kontext [max]", @@ -79,6 +93,8 @@ const Fal: AiProvider = { switch (request.modelId) { case "fal-ai/nano-banana-pro": case "fal-ai/gemini-25-flash-image": + case "fal-ai/flux-2/klein/9b": + case "fal-ai/flux-2/klein/4b": if (genType === "i2i") { endpoint = "/edit"; }