refactor: formatting changes ruff formatter

ruff formatter has slight differences in formatting compared to black
pull/430/head
jaydrennan 5 months ago
parent 7eef3bf628
commit 3322777f5e

@ -715,16 +715,16 @@ class DDPM(pl.LightningModule):
def _TileModeConv2DConvForward(
self,
input_tensor: torch.Tensor,
input: torch.Tensor, # noqa
weight: torch.Tensor,
bias: torch.Tensor, # noqa
bias: torch.Tensor,
):
if self.padding_modeX == self.padding_modeY:
self.padding_mode = self.padding_modeX
return self._orig_conv_forward(input_tensor, weight, bias)
return self._orig_conv_forward(input, weight, bias)
w1 = F.pad(input_tensor, self.paddingX, mode=self.padding_modeX)
del input_tensor
w1 = F.pad(input, self.paddingX, mode=self.padding_modeX)
del input
w2 = F.pad(w1, self.paddingY, mode=self.padding_modeY)
del w1

@ -33,16 +33,16 @@ TileModeType = Literal["", "x", "y", "xy"]
def _tile_mode_conv2d_conv_forward(
self,
tensor_input: torch.Tensor,
input: torch.Tensor, # noqa
weight: torch.Tensor,
bias: torch.Tensor, # noqa
bias: torch.Tensor,
):
if self.padding_mode_x == self.padding_mode_y:
self.padding_mode = self.padding_mode_x
return self._orig_conv_forward(tensor_input, weight, bias)
return self._orig_conv_forward(input, weight, bias)
w1 = F.pad(tensor_input, self.padding_x, mode=self.padding_modeX)
del tensor_input
w1 = F.pad(input, self.padding_x, mode=self.padding_modeX)
del input
w2 = F.pad(w1, self.padding_y, mode=self.padding_modeY)
del w1

Loading…
Cancel
Save