Skip to content

S
Stable Diffusion Webui
Commit cd6c55c1 authored by pepe10-gpu's avatar pepe10-gpu Committed by GitHub
Browse files
Options

16xx card fix 

cudnn
parent 804d9fb8
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
modules/devices.py
View file @ cd6c55c1
...@@ -39,10 +39,13 @@ def torch_gc(): ...@@ -39,10 +39,13 @@ def torch_gc():
def enable_tf32(): def enable_tf32():
if torch.cuda.is_available(): if torch.cuda.is_available():
torch.backends.cudnn.benchmark = True
torch.backends.cudnn.enabled = True
torch.backends.cuda.matmul.allow_tf32 = True torch.backends.cuda.matmul.allow_tf32 = True
torch.backends.cudnn.allow_tf32 = True torch.backends.cudnn.allow_tf32 = True
errors.run(enable_tf32, "Enabling TF32") errors.run(enable_tf32, "Enabling TF32")
device = device_interrogate = device_gfpgan = device_swinir = device_esrgan = device_scunet = device_codeformer = None device = device_interrogate = device_gfpgan = device_swinir = device_esrgan = device_scunet = device_codeformer = None
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment