From 1cf61feeadb67cfc63779cc450196ad40bc7fcca Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Sat, 1 Jul 2023 20:47:11 -0400 Subject: [PATCH] print GPU device at startup --- invokeai/app/services/model_manager_service.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/invokeai/app/services/model_manager_service.py b/invokeai/app/services/model_manager_service.py index 8b46b17ad0..9e5dbc1a40 100644 --- a/invokeai/app/services/model_manager_service.py +++ b/invokeai/app/services/model_manager_service.py @@ -168,6 +168,8 @@ class ModelManagerService(ModelManagerServiceBase): logger.debug(f'config file={config_file}') device = torch.device(choose_torch_device()) + logger.debug(f'GPU device = {device}') + precision = config.precision if precision == "auto": precision = choose_precision(device)