fix the broken line for #10990
This commit is contained in:
parent
b1a72bc7e2
commit
2e23c9c568
@ -48,7 +48,7 @@ class SdOptimizationXformers(SdOptimization):
|
|||||||
priority = 100
|
priority = 100
|
||||||
|
|
||||||
def is_available(self):
|
def is_available(self):
|
||||||
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)
|
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):
|
def apply(self):
|
||||||
ldm.modules.attention.CrossAttention.forward = xformers_attention_forward
|
ldm.modules.attention.CrossAttention.forward = xformers_attention_forward
|
||||||
|
Loading…
Reference in New Issue
Block a user