2 Commits

Author SHA1 Message Date
  w-e-w d4f02b8916
Merge pull request #17129 from namemechan/dev 3 months ago
  namemechan e0be7a8427 update sd_hijack_optimizations.py — Changing [CUDA Compute Capability] Version 3 months ago
1 changed files with 1 additions and 1 deletions
Split View
  1. +1
    -1
      modules/sd_hijack_optimizations.py

+ 1
- 1
modules/sd_hijack_optimizations.py View File

@@ -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…
Cancel
Save
Baidu
map