diff --git a/src/ml/onnx-runtime.d.ts b/src/ml/onnx-runtime.d.ts index ba189df..8cc9d6d 100644 --- a/src/ml/onnx-runtime.d.ts +++ b/src/ml/onnx-runtime.d.ts @@ -23,7 +23,10 @@ declare module "onnxruntime-node" { } export const Tensor: new (type: string, data: unknown, dims: readonly number[]) => OnnxTensor export const InferenceSession: { - create(modelData: ArrayBuffer | Uint8Array): Promise + create( + modelData: ArrayBuffer | Uint8Array, + options?: { intraOpNumThreads?: number }, + ): Promise } }