diff --git a/src/chronos/chronos.py b/src/chronos/chronos.py
index efa6bbf..2beb957 100644
--- a/src/chronos/chronos.py
+++ b/src/chronos/chronos.py
@@ -204,7 +204,10 @@ class ChronosModel(nn.Module):
         super().__init__()
         self.config = config
         self.model = model
-        self.device = model.device
+
+    @property
+    def device(self):
+        return self.model.device
 
     def forward(
         self,