Add method to load model from already loaded state dict

src/spandrel/__helpers/loader.py

@@ -38,7 +38,7 @@ def load_from_file(self, path: str | Path) -> ModelDescriptor:
         """
 
         state_dict = self.load_state_dict_from_file(path)
-        return self.registry.load(state_dict).to(self.device)
+        return self.load_from_state_dict(state_dict)
 
     def load_state_dict_from_file(self, path: str | Path) -> StateDict:
         """
@@ -65,6 +65,15 @@ def load_state_dict_from_file(self, path: str | Path) -> StateDict:
                 f"Unsupported model file extension {extension}. Please try a supported model type."
             )
 
+    def load_from_state_dict(self, state_dict: StateDict) -> ModelDescriptor:
+        """
+        Load a model from the given state dict.
+
+        Throws an `UnsupportedModelError` if the model architecture is not supported.
+        """
+
+        return self.registry.load(state_dict).to(self.device)
+
     def _load_pth(self, path: str | Path) -> StateDict:
         return torch.load(
             path,