Skip to content

Commit e0be7a8

Browse files
committed
update sd_hijack_optimizations.py — Changing [CUDA Compute Capability] Version
1 parent 2174ce5 commit e0be7a8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

modules/sd_hijack_optimizations.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ class SdOptimizationXformers(SdOptimization):
5454
priority = 100
5555

5656
def is_available(self):
57-
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))
57+
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))
5858

5959
def apply(self):
6060
ldm.modules.attention.CrossAttention.forward = xformers_attention_forward

0 commit comments

Comments
 (0)