diff --git a/discriminators.py b/discriminators.py index ffdf327..7dd34aa 100644 --- a/discriminators.py +++ b/discriminators.py @@ -521,7 +521,7 @@ def get_2d_padding( ((kernel_size[1] - 1) * dilation[1]) // 2, ) - def forward(self, x: torch.tensor) -> Tuple[torch.Tensor, List[torch.Tensor]]: + def forward(self, x: torch.Tensor) -> Tuple[torch.Tensor, List[torch.Tensor]]: fmap = [] if self.cqtd_normalize_volume: