react-native-executorch
Version:
An easy way to run AI models in react native with ExecuTorch
18 lines • 708 B
TypeScript
interface Module {
load: (...args: any[]) => Promise<void>;
forward: (...input: any[]) => Promise<any>;
onDownloadProgress: (cb: (progress: number) => void) => void;
}
export declare const useModule: <M extends Module, LoadArgs extends Parameters<M["load"]>, ForwardArgs extends Parameters<M["forward"]>, ForwardReturn extends Awaited<ReturnType<M["forward"]>>>({ module, loadArgs, preventLoad, }: {
module: M;
loadArgs: LoadArgs;
preventLoad?: boolean;
}) => {
error: string | null;
isReady: boolean;
isGenerating: boolean;
downloadProgress: number;
forward: (...input: ForwardArgs) => Promise<ForwardReturn>;
};
export {};
//# sourceMappingURL=useModule.d.ts.map