Skip to content

Commit de44473

Browse files
committed
fix fal ai image-to-image input
1 parent bb963bd commit de44473

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

packages/inference/src/providers/fal-ai.ts

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -240,14 +240,16 @@ export class FalAIImageToImageTask extends FalAiQueueTask implements ImageToImag
240240

241241
async preparePayloadAsync(args: ImageToImageArgs): Promise<RequestArgs> {
242242
const mimeType = args.inputs instanceof Blob ? args.inputs.type : "image/png";
243+
const imageDataUrl = `data:${mimeType};base64,${base64FromBytes(
244+
new Uint8Array(args.inputs instanceof ArrayBuffer ? args.inputs : await (args.inputs as Blob).arrayBuffer())
245+
)}`;
243246
return {
244247
...omit(args, ["inputs", "parameters"]),
245-
image_url: `data:${mimeType};base64,${base64FromBytes(
246-
new Uint8Array(args.inputs instanceof ArrayBuffer ? args.inputs : await (args.inputs as Blob).arrayBuffer())
247-
)}`,
248248
...args.parameters,
249-
...args,
250-
};
249+
image_url: imageDataUrl,
250+
// Some fal endpoints (e.g. FLUX.2-dev) expect `image_urls` (array) instead of `image_url`
251+
image_urls: [imageDataUrl],
252+
} as RequestArgs;
251253
}
252254

253255
override async getResponse(

0 commit comments

Comments
 (0)