Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

print GPU device at startup

Browse Source
This commit is contained in:
Lincoln Stein 2023-07-01 20:47:11 -04:00
parent 5de820f2dc
commit 1cf61feead
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/app/services/model_manager_service.py
View File

@ -168,6 +168,8 @@ class ModelManagerService(ModelManagerServiceBase):
logger.debug(f'config file={config_file}') logger.debug(f'config file={config_file}')
device = torch.device(choose_torch_device()) device = torch.device(choose_torch_device())
logger.debug(f'GPU device = {device}')
precision = config.precision precision = config.precision
if precision == "auto": if precision == "auto":
precision = choose_precision(device) precision = choose_precision(device)