From 276f02d4485346881533edebbd7abebe4d8f359a Mon Sep 17 00:00:00 2001 From: Falko Habel Date: Mon, 2 Jun 2025 18:37:29 +0200 Subject: [PATCH] removed unnessassary parts of config --- src/aiunn/upsampler/aiunn.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/aiunn/upsampler/aiunn.py b/src/aiunn/upsampler/aiunn.py index 978a75d..40ee4e5 100644 --- a/src/aiunn/upsampler/aiunn.py +++ b/src/aiunn/upsampler/aiunn.py @@ -17,11 +17,11 @@ class aiuNN(PreTrainedModel): # Enhanced approach scale_factor = self.config.upsample_scale - out_channels = self.aiia_model.config.num_channels * (scale_factor ** 2) + out_channels = self.config.num_channels * (scale_factor ** 2) self.pixel_shuffle_conv = nn.Conv2d( - in_channels=self.aiia_model.config.hidden_size, + in_channels=self.config.hidden_size, out_channels=out_channels, - kernel_size=self.aiia_model.config.kernel_size, + kernel_size=self.config.kernel_size, padding=1 ) self.pixel_shuffle = nn.PixelShuffle(scale_factor) -- 2.34.1