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

chore: ruff

Browse Source
This commit is contained in:
psychedelicious 2024-03-27 17:32:08 +11:00 committed by Kent Keirsey
parent a291a42abc
commit a397fdbd25
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/backend/util/devices.py
View File

@ -27,10 +27,12 @@ def choose_torch_device() -> torch.device:
else: else:
return torch.device(config.device) return torch.device(config.device)
def get_torch_device_name() -> str: def get_torch_device_name() -> str:
device = choose_torch_device() device = choose_torch_device()
return torch.cuda.get_device_name(device) if device.type == "cuda" else device.type.upper() return torch.cuda.get_device_name(device) if device.type == "cuda" else device.type.upper()
# We are in transition here from using a single global AppConfig to allowing multiple # We are in transition here from using a single global AppConfig to allowing multiple
# configurations. It is strongly recommended to pass the app_config to this function. # configurations. It is strongly recommended to pass the app_config to this function.
def choose_precision( def choose_precision(