ignition-monorepo

ignition-monorepo / backend-onnx/src / loadOnnxModelFromHub

Function: loadOnnxModelFromHub()

loadOnnxModelFromHub(repoId, filename?, maxRetries?): Promise<Buffer<ArrayBufferLike>>

Defined in: backend-onnx/src/io/loadOnnxFromHub.ts:24 

Downloads a .onnx file from HF Hub and returns it as a Buffer. The Buffer can be passed directly to OnnxAgent as modelPath.

Parameters

repoId

string

HF Hub repo (e.g. “salim4n/my-dqn-model”)

filename?

string = 'model.onnx'

Filename in the repo (default: “model.onnx”)

maxRetries?

number = 3

Number of retry attempts on failure (default: 3)

Returns

Promise<Buffer<ArrayBufferLike>>