c-l/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2024-12-29 15:11:58 +00:00
Code Releases Activity
immich/machine-learning/app/models
History
Fynn Petersen-Frey 753292956e
feat(ml): ARMNN acceleration (#5667) 
* feat(ml): ARMNN acceleration for CLIP

* wrap ANN as ONNX-Session

* strict typing

* normalize ARMNN CLIP embedding

* mutex to handle concurrent execution

* make inputs contiguous

* fine-grained locking; concurrent network execution

---------

Co-authored-by: mertalev <101130780+mertalev@users.noreply.github.com>
2024-01-11 18:26:46 +01:00
..
__init__.py feat(server,ml): remove image tagging (#5903) 2023-12-20 20:47:56 -05:00
ann.py feat(ml): ARMNN acceleration (#5667) 2024-01-11 18:26:46 +01:00
base.py feat(ml): ARMNN acceleration (#5667) 2024-01-11 18:26:46 +01:00
cache.py chore(ml): use strict mypy (#5001) 2023-11-13 10:18:46 -06:00
clip.py feat(ml): ARMNN acceleration (#5667) 2024-01-11 18:26:46 +01:00
constants.py Adding the new models to the whitelist (#5736) 2023-12-15 22:45:14 +00:00
facial_recognition.py feat(ml): ARMNN acceleration (#5667) 2024-01-11 18:26:46 +01:00
transforms.py feat(ml): export clip models to ONNX and host models on Hugging Face (#4700) 2023-10-31 05:02:04 -05:00