vndangkhoa/Sys-Arc-Visl
1
0
Fork 0
mirror of https://github.com/vndangkhoa/Sys-Arc-Visl.git synced 2026-04-05 09:27:57 +07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: wrap image in array for processor batch input

Browse source
This commit is contained in:
SysVis AI 2025-12-28 21:14:07 +07:00
parent db777363d7
commit 96670cb678
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
src/lib/visionService.ts
View file

@ -95,9 +95,10 @@ export class VisionService {
// Construct prompts using the processor's method (required for Florence-2) // Construct prompts using the processor's method (required for Florence-2)
const prompts = this.processor.construct_prompts(task); const prompts = this.processor.construct_prompts(task);
// Pre-process the image and text inputs (image first, prompts second) // Pre-process the image and text inputs
// Processor expects batch input, so wrap single image in array
if (!this.processor) throw new Error('Processor is undefined'); if (!this.processor) throw new Error('Processor is undefined');
const inputs = await this.processor(image, prompts); const inputs = await this.processor([image], prompts);
const generatedIds = await this.model.generate({ const generatedIds = await this.model.generate({
...inputs, ...inputs,