From ad27ea2fa2296be8e3f75a8db0e517704122c260 Mon Sep 17 00:00:00 2001
From: Falko Habel <falko.habel@gmx.de>
Date: Sun, 23 Feb 2025 20:04:44 +0100
Subject: [PATCH] fixed to load kwargs

---
 src/aiunn/upsampler.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/aiunn/upsampler.py b/src/aiunn/upsampler.py
index bc48e97..d14133a 100644
--- a/src/aiunn/upsampler.py
+++ b/src/aiunn/upsampler.py
@@ -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)
+        self.config = UpsamplerConfig(kwargs=self.base_model.config)
         # Create the upsample layer using values from the configuration.
         print(self.config.upsample_scale)
         self.upsample = nn.Upsample(