Merge pull request 'removed unnessassary parts of config' (#20) from feat/model_fix into main

Reviewed-on: #20

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)