Skip to content

Commit 3d9f90c

Browse files
authored
fix: multimodal on TGI endpoint type (#1491)
1 parent 8cf6462 commit 3d9f90c

File tree

2 files changed

+4
-12
lines changed

2 files changed

+4
-12
lines changed

src/lib/server/endpoints/tgi/endpointTgi.ts

Lines changed: 3 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,8 @@ export const endpointTgiParametersSchema = z.object({
2323
supportedMimeTypes: ["image/jpeg", "image/webp"],
2424
preferredMimeType: "image/webp",
2525
maxSizeInMB: 5,
26-
maxWidth: 224,
27-
maxHeight: 224,
26+
maxWidth: 378,
27+
maxHeight: 980,
2828
}),
2929
})
3030
.default({}),
@@ -81,22 +81,13 @@ export function endpointTgi(input: z.input<typeof endpointTgiParametersSchema>):
8181
};
8282
}
8383

84-
const whiteImage = {
85-
mime: "image/png",
86-
image: Buffer.from(
87-
"/9j/4AAQSkZJRgABAQAAAQABAAD/2wBDAAEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQH/2wBDAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQEBAQH/wAARCAAQABADAREAAhEBAxEB/8QAHwAAAQUBAQEBAQEAAAAAAAAAAAECAwQFBgcICQoL/8QAtRAAAgEDAwIEAwUFBAQAAAF9AQIDAAQRBRIhMUEGE1FhByJxFDKBkaEII0KxwRVS0fAkM2JyggkKFhcYGRolJicoKSo0NTY3ODk6Q0RFRkdISUpTVFVWV1hZWmNkZWZnaGlqc3R1dnd4eXqDhIWGh4iJipKTlJWWl5iZmqKjpKWmp6ipqrKztLW2t7i5usLDxMXGx8jJytLT1NXW19jZ2uHi4+Tl5ufo6erx8vP09fb3+Pn6/8QAHwEAAwEBAQEBAQEBAQAAAAAAAAECAwQFBgcICQoL/8QAtREAAgECBAQDBAcFBAQAAQJ3AAECAxEEBSExBhJBUQdhcRMiMoEIFEKRobHBCSMzUvAVYnLRChYkNOEl8RcYGRomJygpKjU2Nzg5OkNERUZHSElKU1RVVldYWVpjZGVmZ2hpanN0dXZ3eHl6goOEhYaHiImKkpOUlZaXmJmaoqOkpaanqKmqsrO0tba3uLm6wsPExcbHyMnK0tPU1dbX2Nna4uPk5ebn6Onq8vP09fb3+Pn6/9oADAMBAAIRAxEAPwD+/igAoAKACgD/2Q==",
88-
"base64"
89-
),
90-
};
91-
9284
async function prepareMessage(
9385
isMultimodal: boolean,
9486
message: EndpointMessage,
9587
imageProcessor: ImageProcessor
9688
): Promise<EndpointMessage> {
9789
if (!isMultimodal) return message;
98-
99-
const files = await Promise.all(message.files?.map(imageProcessor) ?? [whiteImage]);
90+
const files = await Promise.all(message.files?.map(imageProcessor) ?? []);
10091
const markdowns = files.map(
10192
(file) => `![](data:${file.mime};base64,${file.image.toString("base64")})`
10293
);

src/lib/server/textGeneration/generate.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ export async function* generate(
1717
continueMessage: isContinue,
1818
generateSettings: assistant?.generateSettings,
1919
toolResults,
20+
isMultimodal: model.multimodal,
2021
})) {
2122
// text generation completed
2223
if (output.generated_text) {

0 commit comments

Comments
 (0)