Skip to content

Commit 28bc95f

Browse files
committed
fix compilation + format
1 parent 1c49625 commit 28bc95f

File tree

4 files changed

+4
-4
lines changed

4 files changed

+4
-4
lines changed

packages/inference/src/lib/makeRequestOptions.ts

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -144,8 +144,7 @@ export async function makeRequestOptions(
144144
? args.data
145145
: JSON.stringify({
146146
...otherArgs,
147-
...(chatCompletion || provider === "together" ? { model } : undefined),
148-
...(provider === "nebius" ? { model } : {}),
147+
...(chatCompletion || provider === "together" || provider === "nebius" ? { model } : undefined),
149148
}),
150149
...(credentials ? { credentials } : undefined),
151150
signal: options?.signal,

packages/inference/src/providers/consts.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@ export const HARDCODED_MODEL_ID_MAPPING: Record<InferenceProvider, Record<ModelI
1919
"fal-ai": {},
2020
"fireworks-ai": {},
2121
"hf-inference": {},
22+
nebius: {},
2223
replicate: {},
2324
sambanova: {},
2425
together: {},

packages/inference/src/tasks/cv/textToImage.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ export async function textToImage(args: TextToImageArgs, options?: Options): Pro
2929
...omit(args, ["inputs", "parameters"]),
3030
...args.parameters,
3131
...(args.provider !== "replicate" ? { response_format: "base64" } : undefined),
32-
...(args.provider === "nebius" ? { response_format: "b64_json" } : {}),
32+
...(args.provider === "nebius" ? { response_format: "b64_json" } : undefined),
3333
prompt: args.inputs,
3434
}
3535
: args;

packages/inference/src/types.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ export type InferenceTask = Exclude<PipelineType, "other">;
4747
export const INFERENCE_PROVIDERS = [
4848
"fal-ai",
4949
"fireworks-ai",
50-
"nebius",
50+
"nebius",
5151
"hf-inference",
5252
"replicate",
5353
"sambanova",

0 commit comments

Comments
 (0)