From 86bb74835d36f5d41e705c487f154e2c74ffa3ca Mon Sep 17 00:00:00 2001
From: Falko Habel <falko.habel@gmx.de>
Date: Sun, 23 Feb 2025 19:59:18 +0100
Subject: [PATCH] updated Upsampler config

---
 src/aiunn/upsampler.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/aiunn/upsampler.py b/src/aiunn/upsampler.py
index fa3b595..efc5c6f 100644
--- a/src/aiunn/upsampler.py
+++ b/src/aiunn/upsampler.py
@@ -1,7 +1,7 @@
 import torch
 import torch.nn as nn
 from aiia import AIIA, AIIAConfig, AIIABase
-
+from config import UpsamplerConfig
 
 # Upsampler model that uses the configuration from the base model.
 class Upsampler(AIIA):
@@ -9,7 +9,7 @@ class Upsampler(AIIA):
         # Assume that base_model.config is an instance of UpsamplerConfig.
         super().__init__(base_model.config)
         self.base_model = base_model
-        
+        self.config = UpsamplerConfig(self.base_model.config)
         # Create the upsample layer using values from the configuration.
         self.upsample = nn.Upsample(
             scale_factor=self.config.upsample_scale,