From 559e156ef7cf7d19fb59dbac6e6bc49543e27cc9 Mon Sep 17 00:00:00 2001 From: RunDevelopment Date: Mon, 20 Nov 2023 17:49:20 +0100 Subject: [PATCH] Remove `state_dict`field --- src/spandrel/__helpers/model_descriptor.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/src/spandrel/__helpers/model_descriptor.py b/src/spandrel/__helpers/model_descriptor.py index 44b46a84..85ac0090 100644 --- a/src/spandrel/__helpers/model_descriptor.py +++ b/src/spandrel/__helpers/model_descriptor.py @@ -74,10 +74,6 @@ def __init__( The specific subclass of `torch.nn.Module` depends on the model architecture. """ - self.state_dict: StateDict = state_dict - """ - The state dict of the model (weights and biases). - """ self.architecture: str = architecture """ The name of the model architecture. E.g. "ESRGAN".