fix: add map_location to torch load

pytorch_tabnet/abstract_model.py

@@ -313,7 +313,7 @@ def load_model(self, filepath):
                     loaded_params = json.load(f)
                 with z.open("network.pt") as f:
                     try:
-                        saved_state_dict = torch.load(f)
+                        saved_state_dict = torch.load(f, map_location=self.device)
                     except io.UnsupportedOperation:
                         # In Python <3.7, the returned file object is not seekable (which at least
                         # some versions of PyTorch require) - so we'll try buffering it in to a