Fixed cuda error by loading AE weights explicitly on CPU

edml/models/provider/path.py

@@ -5,7 +5,7 @@ from torch import nn
 class SerializedModel(nn.Module):
     def __init__(self, model: nn.Module, path: str):
         super().__init__()
-        model.load_state_dict(torch.load(path))
+        model.load_state_dict(torch.load(path, map_location="cpu"))
         self.model = model
 
     def forward(self, x):