spandrel_extra_arches.architectures.SRFormer.SRFormerArch

class documentation

class SRFormerArch(Architecture[SRFormer]): (source)

Undocumented

Method	`__init__`	Undocumented
Method	`load`	Loads the given state dict into a model. The hyperparameters will automatically be deduced.

Inherited from Architecture:

Method	`detect`	Inspects the given state dict and returns `True` if it is a state dict of this architecture.
Property	`id`	The unique ID of the architecture.
Property	`name`	The name of the architecture.
Instance Variable	`_detect`	Undocumented
Instance Variable	`_id`	Undocumented
Instance Variable	`_name`	Undocumented

def __init__(self): (source) ¶

overrides spandrel.Architecture.__init__

Undocumented

@override
def load(self, state_dict: StateDict) -> ImageModelDescriptor[SRFormer]: (source) ¶

overrides spandrel.Architecture.load

Loads the given state dict into a model. The hyperparameters will automatically be deduced.

The state dict is assumed to be a state dict of this architecture, meaning that detect returned True for the state dict. If this is not the case, then the behavior of this function is unspecified (the model may be loaded incorrect or an error is thrown).