mrq/stable-diffusion-webui
1
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

16xx card fix

Browse Source 
cudnn
This commit is contained in:
pepe10-gpu 2022-11-06 17:05:51 -08:00 committed by GitHub
parent 804d9fb83d
commit cd6c55c1ab
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
modules/devices.py
View File

@ -39,10 +39,13 @@ def torch_gc():
def enable_tf32():
if torch.cuda.is_available():
torch.backends.cudnn.benchmark = True
torch.backends.cudnn.enabled = True
torch.backends.cuda.matmul.allow_tf32 = True
torch.backends.cudnn.allow_tf32 = True
errors.run(enable_tf32, "Enabling TF32")
device = device_interrogate = device_gfpgan = device_swinir = device_esrgan = device_scunet = device_codeformer = None