From b25afd5add47ad03a60d576df2fcbde54df57f3b Mon Sep 17 00:00:00 2001
From: grantbuster <grant.buster@nrel.gov>
Date: Tue, 25 Jun 2024 11:23:21 -0600
Subject: [PATCH] save default device to model parameters so model weights can
 be loaded to GPU if requested

---
 sup3r/models/base.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/sup3r/models/base.py b/sup3r/models/base.py
index b7ee1ce26..5ffe17f41 100644
--- a/sup3r/models/base.py
+++ b/sup3r/models/base.py
@@ -351,6 +351,7 @@ def model_params(self):
             'means': means,
             'stdevs': stdevs,
             'meta': self.meta,
+            'default_device': self.default_device,
         }
 
         return model_params