From 9b8cc1bb07c2e4a14c7d9246a85f724cc4476b2b Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Thu, 12 Oct 2023 11:44:43 -0400 Subject: [PATCH] cleanup --- modules/sd_vae_approx.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modules/sd_vae_approx.py b/modules/sd_vae_approx.py index a3c92d955..a12d103ce 100644 --- a/modules/sd_vae_approx.py +++ b/modules/sd_vae_approx.py @@ -9,7 +9,7 @@ sd_vae_approx_model = None class VAEApprox(nn.Module): def __init__(self): - super(VAEApprox, self).__init__() + super().__init__() self.conv1 = nn.Conv2d(4, 8, (7, 7)) self.conv2 = nn.Conv2d(8, 16, (5, 5)) self.conv3 = nn.Conv2d(16, 32, (3, 3)) @@ -64,6 +64,6 @@ def cheap_approximation(sample): ]).reshape(3, 4, 1, 1).to(sample.device) bias = None try: - return nn.functional.conv2d(sample, weight, bias) + return nn.functional.conv2d(sample, weight, bias) # pylint: disable=not-callable except Exception: return sample