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".