Skip to content

Commit b4f30bd

Browse files
Pytorch is stupid. (#10398)
1 parent dad076a commit b4f30bd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

comfy/ops.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ def scaled_dot_product_attention(q, k, v, *args, **kwargs):
5858
NVIDIA_MEMORY_CONV_BUG_WORKAROUND = False
5959
try:
6060
if comfy.model_management.is_nvidia():
61-
if torch.backends.cudnn.version() >= 91200 and comfy.model_management.torch_version_numeric >= (2, 9) and comfy.model_management.torch_version_numeric <= (2, 10):
61+
if torch.backends.cudnn.version() >= 91002 and comfy.model_management.torch_version_numeric >= (2, 9) and comfy.model_management.torch_version_numeric <= (2, 10):
6262
#TODO: change upper bound version once it's fixed'
6363
NVIDIA_MEMORY_CONV_BUG_WORKAROUND = True
6464
logging.info("working around nvidia conv3d memory bug.")

0 commit comments

Comments
 (0)