abc.ABCspandrel.Architecture - The abstract base class for all architectures.spandrel.architectures.ATD.ATDArch - Undocumentedspandrel.architectures.Compact.CompactArch - Undocumentedspandrel.architectures.CRAFT.CRAFTArch - Undocumentedspandrel.architectures.DAT.DATArch - Undocumentedspandrel.architectures.DCTLSA.DCTLSAArch - Undocumentedspandrel.architectures.DITN.DITNArch - Undocumentedspandrel.architectures.DnCNN.DnCNNArch - Undocumentedspandrel.architectures.DRCT.DRCTArch - Undocumentedspandrel.architectures.DRUNet.DRUNetArch - Undocumentedspandrel.architectures.ESRGAN.ESRGANArch - Undocumentedspandrel.architectures.FBCNN.FBCNNArch - Undocumentedspandrel.architectures.FFTformer.FFTformerArch - Undocumentedspandrel.architectures.GFPGAN.GFPGANArch - Undocumentedspandrel.architectures.GRL.GRLArch - Undocumentedspandrel.architectures.HAT.HATArch - Undocumentedspandrel.architectures.HVICIDNet.HVICIDNetArch - Undocumentedspandrel.architectures.IPT.IPTArch - Undocumentedspandrel.architectures.KBNet.KBNetArch - Undocumentedspandrel.architectures.LaMa.LaMaArch - Undocumentedspandrel.architectures.MixDehazeNet.MixDehazeNetArch - Undocumentedspandrel.architectures.MMRealSR.MMRealSRArch - Undocumentedspandrel.architectures.MoESR.MoESRArch - Undocumentedspandrel.architectures.MoSR.MoSRArch - Undocumentedspandrel.architectures.NAFNet.NAFNetArch - Undocumentedspandrel.architectures.OmniSR.OmniSRArch - Undocumentedspandrel.architectures.PLKSR.PLKSRArch - Undocumentedspandrel.architectures.RCAN.RCANArch - Undocumentedspandrel.architectures.RealCUGAN.RealCUGANArch - Undocumentedspandrel.architectures.RestoreFormer.RestoreFormerArch - Undocumentedspandrel.architectures.RetinexFormer.RetinexFormerArch - Undocumentedspandrel.architectures.RGT.RGTArch - Undocumentedspandrel.architectures.SAFMN.SAFMNArch - Undocumentedspandrel.architectures.SAFMNBCIE.SAFMNBCIEArch - Undocumentedspandrel.architectures.SCUNet.SCUNetArch - Undocumentedspandrel.architectures.SeemoRe.SeemoReArch - Undocumentedspandrel.architectures.SPAN.SPANArch - Undocumentedspandrel.architectures.SwiftSRGAN.SwiftSRGANArch - Undocumentedspandrel.architectures.Swin2SR.Swin2SRArch - Undocumentedspandrel.architectures.SwinIR.SwinIRArch - Undocumentedspandrel.architectures.Uformer.UformerArch - Undocumentedspandrel_extra_arches.architectures.AdaCode.AdaCodeArch - Undocumentedspandrel_extra_arches.architectures.DDColor.DDColorArch - Undocumentedspandrel_extra_arches.architectures.FeMaSR.FeMaSRArch - Undocumentedspandrel_extra_arches.architectures.M3SNet.M3SNetArch - Undocumentedspandrel_extra_arches.architectures.MAT.MATArch - Undocumentedspandrel_extra_arches.architectures.MIRNet2.MIRNet2Arch - Undocumentedspandrel_extra_arches.architectures.MPRNet.MPRNetArch - Undocumentedspandrel_extra_arches.architectures.Restormer.RestormerArch - Undocumentedspandrel_extra_arches.architectures.SRFormer.SRFormerArch - Undocumentedspandrel.ModelBase - The base class of all model descriptors.spandrel.ImageModelDescriptor - A model that takes an image as input and returns an image. See __call__ for more information.spandrel.MaskedImageModelDescriptor - A model that takes an image and a mask for that image as input and returns an image. See __call__ for more information.enum.Enumspandrel.ModelTiling - Describes whether and how a model supports tiling.Exceptionspandrel.UnsupportedDtypeError - An error that will be thrown by .to if the model does not support the given dtype.spandrel.UnsupportedModelError - An error that will be thrown by ArchRegistry and ModelLoader if a model architecture is not supported.spandrel.ArchRegistry - A registry of architectures.spandrel.ArchSupport - An entry in an ArchRegistry that describes how to detect and load a model architecture.spandrel.ModelLoader - Class for automatically loading a pth file into any architecturespandrel.SizeRequirements - A set of requirements for the size of an input image.spandrel.util.KeyCondition - A condition that checks if a state dict has the given keys.spandrel_extra_arches.architectures.CodeFormer.__arch.codeformer.VQAutoEncoderspandrel_extra_arches.architectures.CodeFormer.CodeFormer - Undocumentedtorch.nn.Modulespandrel.architectures.__arch_helpers.block.ConcatBlock - Undocumentedspandrel.architectures.__arch_helpers.block.ResidualDenseBlock_5C - Residual Dense Block Style: 5 convs The core module of paper: (Residual Dense Network for Image Super-Resolution, CVPR 18) Modified options that can be used:spandrel.architectures.__arch_helpers.block.ResNetBlock - ResNet Block, 3-3 style with extra residual scaling used in EDSR (Enhanced Deep Residual Networks for Single Image Super-Resolution, CVPRW 17)spandrel.architectures.__arch_helpers.block.RRDB - Residual in Residual Dense Block (ESRGAN: Enhanced Super-Resolution Generative Adversarial Networks)spandrel.architectures.__arch_helpers.block.ShortcutBlock - Undocumentedspandrel.architectures.__arch_helpers.dysample.DySample - Adapted from 'Learning to Upsample by Learning to Sample': https://arxiv.org/abs/2308.15085 https://github.com/tiny-smart/dysamplespandrel.architectures.Compact.Compact - A compact VGG-style network structure for super-resolution.spandrel.architectures.CRAFT.CRAFT - Some codes are based on SwinIR.spandrel.architectures.DAT.DAT - Dual Aggregation Transformerspandrel.architectures.DCTLSA.DCTLSA - Undocumentedspandrel.architectures.DITN.DITN - Undocumentedspandrel.architectures.DnCNN.DnCNN - No class docstring; 0/2 instance variable, 0/1 class variable, 1/2 method documentedspandrel.architectures.DRCT.DRCT - Undocumentedspandrel.architectures.DRUNet.DRUNet - Undocumentedspandrel.architectures.ESRGAN.ESRGAN - No class docstring; 0/3 instance variable, 0/1 class variable, 1/2 method documentedspandrel.architectures.FBCNN.FBCNN - Undocumentedspandrel.architectures.FFTformer.FFTformer - Undocumentedspandrel.architectures.GFPGAN.GFPGAN - The GFPGAN architecture: Unet + StyleGAN2 decoder with SFT. It is the clean version without custom compiled CUDA extensions used in StyleGAN2. Ref: GFP-GAN: Towards Real-World Blind Face Restoration with Generative Facial Prior.spandrel.architectures.GRL.GRL - Image restoration transformer with global, non-local, and local connectionsspandrel.architectures.HAT.HAT - A PyTorch implementation of : Activating More Pixels in Image Super-Resolution Transformer. Some codes are based on SwinIR.spandrel.architectures.HVICIDNet.HVICIDNet - Undocumentedspandrel.architectures.IPT.IPT - Undocumentedspandrel.architectures.KBNet.KBNet_l - Undocumentedspandrel.architectures.KBNet.KBNet_s - Undocumentedspandrel.architectures.LaMa.LaMa - Undocumentedspandrel.architectures.MixDehazeNet.MixDehazeNet - Undocumentedspandrel.architectures.MMRealSR.MMRealSR - Networks consisting of Residual in Residual Dense Block, which is used in ESRGAN. v2.1spandrel.architectures.MoSR.MoSR - Mamba Out Super-Resolutionspandrel.architectures.NAFNet.NAFNet - Undocumentedspandrel.architectures.OmniSR.OmniSR - Undocumentedspandrel.architectures.PLKSR.PLKSR - Undocumentedspandrel.architectures.PLKSR.RealPLKSR - Partial Large Kernel CNNs for Efficient Super-Resolution: https://arxiv.org/abs/2404.11848spandrel.architectures.RealCUGAN.UpCunet2x - Undocumentedspandrel.architectures.RealCUGAN.UpCunet2x_fast - Undocumentedspandrel.architectures.RealCUGAN.UpCunet3x - Undocumentedspandrel.architectures.RealCUGAN.UpCunet4x - Undocumentedspandrel.architectures.RestoreFormer.RestoreFormer - Undocumentedspandrel.architectures.RetinexFormer.RetinexFormer - No class docstring; 0/2 instance variable, 0/1 class variable, 1/2 method documentedspandrel.architectures.RGT.RGT - No class docstring; 0/15 instance variable, 0/1 class variable, 1/3 method documentedspandrel.architectures.SAFMN.SAFMN - Undocumentedspandrel.architectures.SAFMNBCIE.SAFMNBCIE - Undocumentedspandrel.architectures.SCUNet.SCUNet - Undocumentedspandrel.architectures.SeemoRe.SeemoRe - Undocumentedspandrel.architectures.SPAN.SPAN - Swift Parameter-free Attention Network for Efficient Super-Resolutionspandrel.architectures.SwiftSRGAN.SwiftSRGAN - Swift-SRGAN Generatorspandrel.architectures.Swin2SR.Swin2SR - A PyTorch impl of : Swin2SR: SwinV2 Transformer for Compressed Image Super-Resolution and Restoration.spandrel.architectures.SwinIR.SwinIR - A PyTorch impl of : SwinIR: Image Restoration Using Swin Transformer, based on Swin Transformer.spandrel.architectures.Uformer.Uformer - Undocumentedspandrel.util.timm.DropBlock2d - DropBlock. See https://arxiv.org/pdf/1810.12890.pdfspandrel.util.timm.DropPath - Drop paths (Stochastic Depth) per sample (when applied in main path of residual blocks).spandrel_extra_arches.architectures.AdaCode.AdaCode - Undocumentedspandrel_extra_arches.architectures.DDColor.DDColor - Undocumentedspandrel_extra_arches.architectures.FeMaSR.FeMaSR - Undocumentedspandrel_extra_arches.architectures.M3SNet.M3SNet - Undocumentedspandrel_extra_arches.architectures.MAT.MAT - No class docstring; 0/3 instance variable, 0/1 class variable, 1/2 method documentedspandrel_extra_arches.architectures.MIRNet2.MIRNet2 - Undocumentedspandrel_extra_arches.architectures.MPRNet.MPRNet - Undocumentedspandrel_extra_arches.architectures.Restormer.Restormer - Undocumentedspandrel_extra_arches.architectures.SRFormer.SRFormer - A PyTorch implement of : SRFormer: Permuted Self-Attention for Single Image Super-Resolution, based on Swin Transformer.typing.Genericspandrel.Architecture - The abstract base class for all architectures.spandrel.architectures.ATD.ATDArch - Undocumentedspandrel.architectures.Compact.CompactArch - Undocumentedspandrel.architectures.CRAFT.CRAFTArch - Undocumentedspandrel.architectures.DAT.DATArch - Undocumentedspandrel.architectures.DCTLSA.DCTLSAArch - Undocumentedspandrel.architectures.DITN.DITNArch - Undocumentedspandrel.architectures.DnCNN.DnCNNArch - Undocumentedspandrel.architectures.DRCT.DRCTArch - Undocumentedspandrel.architectures.DRUNet.DRUNetArch - Undocumentedspandrel.architectures.ESRGAN.ESRGANArch - Undocumentedspandrel.architectures.FBCNN.FBCNNArch - Undocumentedspandrel.architectures.FFTformer.FFTformerArch - Undocumentedspandrel.architectures.GFPGAN.GFPGANArch - Undocumentedspandrel.architectures.GRL.GRLArch - Undocumentedspandrel.architectures.HAT.HATArch - Undocumentedspandrel.architectures.HVICIDNet.HVICIDNetArch - Undocumentedspandrel.architectures.IPT.IPTArch - Undocumentedspandrel.architectures.KBNet.KBNetArch - Undocumentedspandrel.architectures.LaMa.LaMaArch - Undocumentedspandrel.architectures.MixDehazeNet.MixDehazeNetArch - Undocumentedspandrel.architectures.MMRealSR.MMRealSRArch - Undocumentedspandrel.architectures.MoESR.MoESRArch - Undocumentedspandrel.architectures.MoSR.MoSRArch - Undocumentedspandrel.architectures.NAFNet.NAFNetArch - Undocumentedspandrel.architectures.OmniSR.OmniSRArch - Undocumentedspandrel.architectures.PLKSR.PLKSRArch - Undocumentedspandrel.architectures.RCAN.RCANArch - Undocumentedspandrel.architectures.RealCUGAN.RealCUGANArch - Undocumentedspandrel.architectures.RestoreFormer.RestoreFormerArch - Undocumentedspandrel.architectures.RetinexFormer.RetinexFormerArch - Undocumentedspandrel.architectures.RGT.RGTArch - Undocumentedspandrel.architectures.SAFMN.SAFMNArch - Undocumentedspandrel.architectures.SAFMNBCIE.SAFMNBCIEArch - Undocumentedspandrel.architectures.SCUNet.SCUNetArch - Undocumentedspandrel.architectures.SeemoRe.SeemoReArch - Undocumentedspandrel.architectures.SPAN.SPANArch - Undocumentedspandrel.architectures.SwiftSRGAN.SwiftSRGANArch - Undocumentedspandrel.architectures.Swin2SR.Swin2SRArch - Undocumentedspandrel.architectures.SwinIR.SwinIRArch - Undocumentedspandrel.architectures.Uformer.UformerArch - Undocumentedspandrel_extra_arches.architectures.AdaCode.AdaCodeArch - Undocumentedspandrel_extra_arches.architectures.DDColor.DDColorArch - Undocumentedspandrel_extra_arches.architectures.FeMaSR.FeMaSRArch - Undocumentedspandrel_extra_arches.architectures.M3SNet.M3SNetArch - Undocumentedspandrel_extra_arches.architectures.MAT.MATArch - Undocumentedspandrel_extra_arches.architectures.MIRNet2.MIRNet2Arch - Undocumentedspandrel_extra_arches.architectures.MPRNet.MPRNetArch - Undocumentedspandrel_extra_arches.architectures.Restormer.RestormerArch - Undocumentedspandrel_extra_arches.architectures.SRFormer.SRFormerArch - Undocumentedspandrel.ImageModelDescriptor - A model that takes an image as input and returns an image. See __call__ for more information.spandrel.MaskedImageModelDescriptor - A model that takes an image and a mask for that image as input and returns an image. See __call__ for more information.spandrel.ModelBase - The base class of all model descriptors.spandrel.ImageModelDescriptor - A model that takes an image as input and returns an image. See __call__ for more information.spandrel.MaskedImageModelDescriptor - A model that takes an image and a mask for that image as input and returns an image. See __call__ for more information.ValueErrorspandrel.DuplicateArchitectureError - An error that will be thrown by ArchRegistry if the same architecture is added twice.