mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2026-04-21 04:11:01 -07:00
update sd_hijack_optimizations.py — Changing [CUDA Compute Capability] Version
This commit is contained in:
parent
2174ce5afe
commit
e0be7a8427
1 changed files with 1 additions and 1 deletions
|
|
@ -54,7 +54,7 @@ class SdOptimizationXformers(SdOptimization):
|
|||
priority = 100
|
||||
|
||||
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) <= (12, 0))
|
||||
|
||||
def apply(self):
|
||||
ldm.modules.attention.CrossAttention.forward = xformers_attention_forward
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue