wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-03 21:20:49 +08:00
Code Issues Actions 22 Packages Projects Releases Wiki Activity

Remove SeedVR2 VAE memory convolution workaround

Browse Source
This commit is contained in:
John Pollock 2026-07-01 07:07:41 -05:00
parent ad04a6199e
commit e595965392
1 changed files with 0 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
comfy/ldm/seedvr/vae.py
View File

@ -30,7 +30,6 @@ from comfy.ldm.modules.diffusionmodules.model import vae_attention
import math
from enum import Enum
from comfy.ops import NVIDIA_MEMORY_CONV_BUG_WORKAROUND
import logging
import comfy.model_management
@ -597,23 +596,6 @@ class InflatedCausalConv3d(ops.Conv3d):
self.memory_limit = value
def _conv_forward(self, input, weight, bias, *args, **kwargs):
if (NVIDIA_MEMORY_CONV_BUG_WORKAROUND and
weight.dtype in (torch.float16, torch.bfloat16) and
hasattr(torch.backends.cudnn, 'is_available') and
torch.backends.cudnn.is_available() and
getattr(torch.backends.cudnn, 'enabled', True)):
try:
out = torch.cudnn_convolution(
input, weight, self.padding, self.stride, self.dilation, self.groups,
benchmark=False, deterministic=False, allow_tf32=True
)
if bias is not None:
out += bias.reshape((1, -1) + (1,) * (out.ndim - 2))
return out
except RuntimeError:
pass
except NotImplementedError:
pass
try:
return super()._conv_forward(input, weight, bias, *args, **kwargs)
except NotImplementedError: