qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2026-07-26 05:29:16 +02:00
Code Issues Projects Releases Wiki Activity

ModelZoo: avoid Azure EP (displacing CPU)

Browse source
This commit is contained in:
Robert Sachunsky 2026-07-15 04:09:33 +02:00
parent b4165114ea
commit 839d7b9a7e
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/eynollah/model_zoo/model_zoo.py
View file

@ -334,6 +334,8 @@ class EynollahModelZoo:
providers = [provider for provider in providers providers = [provider for provider in providers
if provider[:-17] in override_providers] if provider[:-17] in override_providers]
# configure and prioritise # configure and prioritise
if 'AzureExecutionProvider' in providers:
providers.remove('AzureExecutionProvider')
if 'CUDAExecutionProvider' in providers: if 'CUDAExecutionProvider' in providers:
providers.remove('CUDAExecutionProvider') providers.remove('CUDAExecutionProvider')
if gpu >= 0: if gpu >= 0: