From d2159af10e27c3783676f1e2384ad00f4bf18e7e Mon Sep 17 00:00:00 2001 From: Disty0 Date: Mon, 26 May 2025 04:24:28 +0300 Subject: [PATCH] cleanup --- modules/model_quant_sdnq.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modules/model_quant_sdnq.py b/modules/model_quant_sdnq.py index 0cfa48dfa..c53a757ec 100644 --- a/modules/model_quant_sdnq.py +++ b/modules/model_quant_sdnq.py @@ -501,8 +501,8 @@ def quantized_conv_transpose_2d_forward(self, input: torch.FloatTensor, output_s def quantized_conv_transpose_3d_forward(self, input: torch.FloatTensor, output_size: Optional[list[int]] = None) -> torch.FloatTensor: - output_padding = self._output_padding(input, output_size, self.stride, self.padding, self.kernel_size, 3, self.dilation) - return torch.nn.functional.conv_transpose3d(input, self.sdnq_decompressor(self, return_decompressed_only=True), self.bias, self.stride, self.padding, output_padding, self.groups, self.dilation) + output_padding = self._output_padding(input, output_size, self.stride, self.padding, self.kernel_size, 3, self.dilation) + return torch.nn.functional.conv_transpose3d(input, self.sdnq_decompressor(self, return_decompressed_only=True), self.bias, self.stride, self.padding, output_padding, self.groups, self.dilation) class INT8AsymmetricWeightsDecompressor(torch.nn.Module):