react-native-executorch/lib/module/hooks/natural_language_processing/useTextEmbeddings.js

An easy way to run AI models in React Native with ExecuTorch

"use strict";

import { TextEmbeddingsModule } from '../../modules/natural_language_processing/TextEmbeddingsModule';
import { useModuleFactory } from '../useModuleFactory';
/**
 * React hook for managing a Text Embeddings model instance.
 * @category Hooks
 * @param TextEmbeddingsProps - Configuration object containing `model` source and optional `preventLoad` flag.
 * @returns Ready to use Text Embeddings model.
 */
export const useTextEmbeddings = ({
  model,
  preventLoad = false
}) => {
  const {
    error,
    isReady,
    isGenerating,
    downloadProgress,
    runForward
  } = useModuleFactory({
    factory: (config, onProgress) => TextEmbeddingsModule.fromModelName(config, onProgress),
    config: model,
    deps: [model.modelName, model.modelSource, model.tokenizerSource],
    preventLoad
  });
  const forward = input => runForward(inst => inst.forward(input));
  return {
    error,
    isReady,
    isGenerating,
    downloadProgress,
    forward
  };
};
//# sourceMappingURL=useTextEmbeddings.js.map