diff --git a/packages/inference/src/providers/hf-inference.ts b/packages/inference/src/providers/hf-inference.ts index 07b7a3c3d..b049c0b54 100644 --- a/packages/inference/src/providers/hf-inference.ts +++ b/packages/inference/src/providers/hf-inference.ts @@ -303,7 +303,12 @@ export class HFInferenceImageSegmentationTask extends HFInferenceTask implements override async getResponse(response: ImageSegmentationOutput): Promise { if ( Array.isArray(response) && - response.every((x) => typeof x.label === "string" && typeof x.mask === "string" && typeof x.score === "number") + response.every( + (x) => + typeof x.label === "string" && + typeof x.mask === "string" && + (x.score === undefined || typeof x.score === "number") + ) ) { return response; }