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