Method __init__

diffsynth/models/cog_vae.py:170–172

(self, in_channels, out_channels, kernel_size, stride=1, padding=0)

class CachedConv3d(torch.nn.Conv3d):
    def __init__(self, in_channels, out_channels, kernel_size, stride=1, padding=0):
        super().__init__(in_channels, out_channels, kernel_size=kernel_size, stride=stride, padding=padding)
        self.cached_tensor = None


    def clear_cache(self):