Skip to content

S
Stable Diffusion Webui
Commit b1a72bc7 authored by Vivek K. Vasishtha's avatar Vivek K. Vasishtha Committed by GitHub
Browse files
Options

torch.cuda.is_available() check for SdOptimizationXformers

parent b6af0a38
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
modules/sd_hijack_optimizations.py
View file @ b1a72bc7
......@@ -48,7 +48,7 @@ class SdOptimizationXformers(SdOptimization):
priority = 100
def is_available(self):
return shared.cmd_opts.force_enable_xformers or (shared.xformers_available and torch.version.cuda and (6, 0) <= torch.cuda.get_device_capability(shared.device) <= (9, 0))
return shared.cmd_opts.force_enable_xformers or (shared.xformers_available and torch.cuda.is_available() and (6, 0) <= torch.cuda.get_device_capability(shared.device) <= (9, 0)
def apply(self):
ldm.modules.attention.CrossAttention.forward = xformers_attention_forward
......
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