Backupppert/immich
1
1
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2025-07-13 20:38:46 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions1

feat(ml): improved ARM-NN support ()

Browse source
This commit is contained in:
Fynn Petersen-Frey 2024-07-20 21:59:27 +02:00 • committed by GitHub
parent 7c3326b662
commit 54488b1016
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 70 additions and 32 deletions
machine-learning/app
test_main.py

4
machine-learning/app/test_main.py
View file

@ -268,9 +268,9 @@ class TestAnnSession:
AnnSession(model_path, cache_dir)
ann_session.assert_called_once_with(tuning_level=3, tuning_file=(cache_dir / "gpu-tuning.ann").as_posix())
ann_session.assert_called_once_with(tuning_level=2, tuning_file=(cache_dir / "gpu-tuning.ann").as_posix())
ann_session.return_value.load.assert_called_once_with(
model_path.as_posix(), cached_network_path=model_path.with_suffix(".anncache").as_posix()
model_path.as_posix(), cached_network_path=model_path.with_suffix(".anncache").as_posix(), fp16=False
)
info.assert_has_calls(
[