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

Update to exclude GTX 16xx from autocast selection (#576)

Browse Source 
* Update to exclude GTX 1660/1650 from autocast selection
This commit is contained in:
ArDiouscuros 2022-09-16 01:29:19 +02:00 committed by GitHub
parent e6011631a1
commit f9b272a7b9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
ldm/dream/devices.py
View File

@ -13,9 +13,12 @@ def choose_torch_device() -> str:
def choose_autocast_device(device): def choose_autocast_device(device):
'''Returns an autocast compatible device from a torch device''' '''Returns an autocast compatible device from a torch device'''
device_type = device.type # this returns 'mps' on M1 device_type = device.type # this returns 'mps' on M1
# autocast only for cuda, but GTX 16xx have issues with it
if device_type == 'cuda': if device_type == 'cuda':
return device_type,autocast device_name = torch.cuda.get_device_name()
elif device_type == 'cpu': if 'GeForce GTX 1660' in device_name or 'GeForce GTX 1650' in device_name:
return device_type,nullcontext return device_type,nullcontext
else:
return device_type,autocast
else: else:
return 'cpu',nullcontext return 'cpu',nullcontext