Corrected one line only to have the second one be incorrect too

mala/network/runner.py

@@ -852,7 +852,7 @@ def _forward_entire_snapshot(
         # activated AFTER loading a model.
         if self.parameters_full.use_ddp:
             self.network.module.to(
-                self.network.params._configuration["device"]
+                self.network.module.params._configuration["device"]
             )
         else:
             self.network.to(self.network.params._configuration["device"])