public/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2025-11-17 12:52:38 +09:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat(ml): add cache_dir option to OpenVINO EP (#8018)

Browse Source 
* add cache_dir option to OpenVINO EP

* update provider options test to include cache_dir

* use forward slash instead of string concatenation

* fix cache_dir placement in provider options assertion
This commit is contained in:
Davide
2024-03-17 18:48:59 +01:00
committed by GitHub
parent b8278404a0
commit e8f0f82db0
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
machine-learning/app/test_main.py
View File

@@ -88,7 +88,7 @@ class TestBase:
encoder = OpenCLIPEncoder("ViT-B-32__openai", providers=["OpenVINOExecutionProvider", "CPUExecutionProvider"])
assert encoder.provider_options == [
{"device_type": "GPU_FP32"},
{"device_type": "GPU_FP32", "cache_dir": (encoder.cache_dir / "openvino").as_posix()},
{"arena_extend_strategy": "kSameAsRequested"},
]