This commit is contained in:
Manav Rathi 2024-04-11 15:06:03 +05:30
parent 52727f2255
commit f5bf776848
No known key found for this signature in database
2 changed files with 7 additions and 6 deletions

View file

@ -1,8 +1,8 @@
# Dependencies
* [Electron](#electron)
* [Dev dependencies](#dev)
* [Functionality](#functionality)
- [Electron](#electron)
- [Dev dependencies](#dev)
- [Functionality](#functionality)
## Electron
@ -114,8 +114,8 @@ available on the host machine, and is not bundled with our app.
AI/ML runtime. It powers both natural language searches (using CLIP) and face
detection (using YOLO).
[jpeg-js](https://github.com/jpeg-js/jpeg-js#readme) is used for decoding
JPEG data into raw RGB bytes before passing it to ONNX.
[jpeg-js](https://github.com/jpeg-js/jpeg-js#readme) is used for decoding JPEG
data into raw RGB bytes before passing it to ONNX.
html-entities is used by the bundled clip-bpe-ts tokenizer for CLIP.

View file

@ -139,5 +139,6 @@ export const faceEmbedding = async (input: Float32Array) => {
const feeds = { img_inputs: inputTensor };
const results = await session.run(feeds);
log.debug(() => `onnx/yolo face embedding took ${Date.now() - t} ms`);
return results.embeddings["cpuData"]; // as Float32Array;
// TODO: What's with this type?
return (results.embeddings as unknown as any)["cpuData"]; // as Float32Array;
};