From 6a5528cccd080a3667577ff244e7a8b1dd6072f0 Mon Sep 17 00:00:00 2001
From: Brendan Lackey <brendan.lackey@outlook.com>
Date: Thu, 21 Nov 2024 15:07:16 -0700
Subject: [PATCH] Add some doc comments to classes

---
 hidiffusion/raunet.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/hidiffusion/raunet.py b/hidiffusion/raunet.py
index 66a7074..ca8b53e 100644
--- a/hidiffusion/raunet.py
+++ b/hidiffusion/raunet.py
@@ -72,7 +72,7 @@ class HDUpsample(ORIG_UPSAMPLE):
             Returns:
                 torch.Tensor: Upsampled tensor after processing through interpolation and convolution
     """
-    
+
     def forward(self, x, output_shape=None, transformer_options=None):
         if self.dims == 3 or not self.use_conv or not HDCONFIG.check(transformer_options):
             return super().forward(x, output_shape=output_shape)