From 60c9f4786c6668fe8aa8a8eaa4a78d4088481642 Mon Sep 17 00:00:00 2001 From: Robert Sachunsky Date: Fri, 12 Jun 2026 14:58:32 +0200 Subject: [PATCH] =?UTF-8?q?ModelZoo=20device=20selection:=20warn=20if=20mo?= =?UTF-8?q?del=20category=20still=20unmatched=E2=80=A6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit (and try GPU) --- src/eynollah/model_zoo/model_zoo.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/src/eynollah/model_zoo/model_zoo.py b/src/eynollah/model_zoo/model_zoo.py index d028004..13656f4 100644 --- a/src/eynollah/model_zoo/model_zoo.py +++ b/src/eynollah/model_zoo/model_zoo.py @@ -182,7 +182,10 @@ class EynollahModelZoo: if fnmatchcase(model_category, cat): device = dev break - if device == 'CPU': + if ':' in device: + self.logger.warning("missing device specification for model type %s", model_category) + gpus = gpus[:1] + elif device == 'CPU': gpus = [] else: assert device.startswith('GPU') @@ -227,6 +230,9 @@ class EynollahModelZoo: if fnmatchcase('ocr', cat): device = dev break + if ':' in device: + self.logger.warning("missing device specification for model type %s", model_category) + device = 'GPU' if device and device.startswith('GPU'): try: device0 = torch.device('cuda', int(device[3:] or 0)) @@ -309,7 +315,10 @@ class EynollahModelZoo: if fnmatchcase(model_category, cat): device = dev break - if device == 'CPU': + if ':' in device: + self.logger.warning("missing device specification for model type %s", model_category) + gpu = 0 + elif device == 'CPU': gpu = -1 else: assert device.startswith('GPU')