Mirror/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2024-06-07 21:20:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix #4407 breaking UI entirely for card other than ones related to the PR

Browse Source
This commit is contained in:
AUTOMATIC 2022-12-03 15:57:52 +03:00
parent ce049c471b
commit 2651267e3a
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
modules/devices.py
View File

@ -53,12 +53,10 @@ def torch_gc():
def enable_tf32():
if torch.cuda.is_available():
for devid in range(0,torch.cuda.device_count()):
if torch.cuda.get_device_capability(devid) == (7, 5):
shd = True
if shd:
if any([torch.cuda.get_device_capability(devid) == (7, 5) for devid in range(0, torch.cuda.device_count())]):
torch.backends.cudnn.benchmark = True
torch.backends.cudnn.enabled = True
torch.backends.cuda.matmul.allow_tf32 = True
torch.backends.cudnn.allow_tf32 = True