chaiNNer-org · joeyballentine · Jul 7, 2024 · Jul 7, 2024
diff --git a/CONTRIBUTING.md b/CONTRIBUTING.md
@@ -110,11 +110,11 @@ Of course, the copy this code, we first have to find it! Unfortunately, project
 
 In the case of DITN, the file is called [`models/DITN_Real.py`](https://github.com/yongliuy/DITN/blob/3438e429c0538ee5061a7cfca587df0c4097703f/models/DITN_Real.py#L197).
 
-Once you have found the file, create a new directory `libs/spandrel/architectures/<arch name>/arch/` and copy the file into it. This directory will contain all code that we will copy. Since we respect the copy right of the original authors, we will also copy the `LICENSE` file of the repo into this directory.
+Once you have found the file, create a new directory `libs/spandrel/architectures/<arch name>/__arch/` and copy the file into it. This directory will contain all code that we will copy. Since we respect the copy right of the original authors, we will also copy the `LICENSE` file of the repo into this directory.
 
-In the case of DITN, we copy `models/DITN_Real.py` to `libs/spandrel/architectures/DITN/arch/DITN_Real.py` and add the `LICENSE` file of the repo.
+In the case of DITN, we copy `models/DITN_Real.py` to `libs/spandrel/architectures/DITN/__arch/DITN_Real.py` and add the `LICENSE` file of the repo.
 
-The main model file might also reference other files in the repo. You have to copy those files into your `arch/` directory as well.
+The main model file might also reference other files in the repo. You have to copy those files into your `__arch/` directory as well.
 
 In the case of DITN, the main model file doesn't reference any other files, so we are done.
 
@@ -160,7 +160,7 @@ Create a file `libs/spandrel/architectures/<arch name>/__init__.py` and add the
 
 ```python
 from ...__helpers.model_descriptor import ImageModelDescriptor, StateDict
-from .arch.ARCH_NAME import ARCH_NAME
+from .__arch.ARCH_NAME import ARCH_NAME
 
 
 def load(state_dict: StateDict) -> ImageModelDescriptor[ARCH_NAME]:
@@ -205,7 +205,7 @@ In the case of DITN, the filled in template looks like this:
 
 ```python
 from ...__helpers.model_descriptor import ImageModelDescriptor, StateDict
-from .arch.DITN_Real import DITN_Real as DITN
+from .__arch.DITN_Real import DITN_Real as DITN
 
 
 def load(state_dict: StateDict) -> ImageModelDescriptor[DITN]:

diff --git a/README.md b/README.md
@@ -169,4 +169,4 @@ Here are some cool projects that use Spandrel:
 
 ## License
 
-This repo is bounded by the MIT license. However, the code of implemented architectures (everything inside an `arch/` directory) is bound by their original respective licenses (which are included in their respective `arch/` directories).
+This repo is bounded by the MIT license. However, the code of implemented architectures (everything inside an `__arch/` directory) is bound by their original respective licenses (which are included in their respective `__arch/` directories).
diff --git a/...l/spandrel/architectures/ATD/arch/LICENSE → ...spandrel/architectures/ATD/__arch/LICENSE b/...l/spandrel/architectures/ATD/arch/LICENSE → ...spandrel/architectures/ATD/__arch/LICENSE
diff --git a/...andrel/architectures/ATD/arch/atd_arch.py → ...drel/architectures/ATD/__arch/atd_arch.py b/...andrel/architectures/ATD/arch/atd_arch.py → ...drel/architectures/ATD/__arch/atd_arch.py
diff --git a/libs/spandrel/spandrel/architectures/ATD/__init__.py b/libs/spandrel/spandrel/architectures/ATD/__init__.py
@@ -10,7 +10,7 @@
     SizeRequirements,
     StateDict,
 )
-from .arch.atd_arch import ATD
+from .__arch.atd_arch import ATD
 
 
 class ATDArch(Architecture[ATD]):

diff --git a/...pandrel/architectures/CRAFT/arch/CRAFT.py → ...ndrel/architectures/CRAFT/__arch/CRAFT.py b/...pandrel/architectures/CRAFT/arch/CRAFT.py → ...ndrel/architectures/CRAFT/__arch/CRAFT.py
diff --git a/...spandrel/architectures/CRAFT/arch/LICENSE → ...andrel/architectures/CRAFT/__arch/LICENSE b/...spandrel/architectures/CRAFT/arch/LICENSE → ...andrel/architectures/CRAFT/__arch/LICENSE
diff --git a/libs/spandrel/spandrel/architectures/CRAFT/__init__.py b/libs/spandrel/spandrel/architectures/CRAFT/__init__.py
@@ -10,7 +10,7 @@
     SizeRequirements,
     StateDict,
 )
-from .arch.CRAFT import CRAFT
+from .__arch.CRAFT import CRAFT
 
 
 class CRAFTArch(Architecture[CRAFT]):

diff --git a/...andrel/architectures/Compact/arch/LICENSE → ...drel/architectures/Compact/__arch/LICENSE b/...andrel/architectures/Compact/arch/LICENSE → ...drel/architectures/Compact/__arch/LICENSE
diff --git a/...ndrel/architectures/Compact/arch/SRVGG.py → ...rel/architectures/Compact/__arch/SRVGG.py b/...ndrel/architectures/Compact/arch/SRVGG.py → ...rel/architectures/Compact/__arch/SRVGG.py
diff --git a/libs/spandrel/spandrel/architectures/Compact/__init__.py b/libs/spandrel/spandrel/architectures/Compact/__init__.py
@@ -3,7 +3,7 @@
 from spandrel.util import KeyCondition, get_scale_and_output_channels, get_seq_len
 
 from ...__helpers.model_descriptor import Architecture, ImageModelDescriptor, StateDict
-from .arch.SRVGG import SRVGGNetCompact
+from .__arch.SRVGG import SRVGGNetCompact
 
 
 class CompactArch(Architecture[SRVGGNetCompact]):

diff --git a/...el/spandrel/architectures/DAT/arch/DAT.py → .../spandrel/architectures/DAT/__arch/DAT.py b/...el/spandrel/architectures/DAT/arch/DAT.py → .../spandrel/architectures/DAT/__arch/DAT.py
diff --git a/...l/spandrel/architectures/DAT/arch/LICENSE → ...spandrel/architectures/DAT/__arch/LICENSE b/...l/spandrel/architectures/DAT/arch/LICENSE → ...spandrel/architectures/DAT/__arch/LICENSE
diff --git a/libs/spandrel/spandrel/architectures/DAT/__init__.py b/libs/spandrel/spandrel/architectures/DAT/__init__.py
@@ -10,7 +10,7 @@
     SizeRequirements,
     StateDict,
 )
-from .arch.DAT import DAT
+from .__arch.DAT import DAT
 
 
 class DATArch(Architecture[DAT]):

diff --git a/...pandrel/architectures/DCTLSA/arch/LICENSE → ...ndrel/architectures/DCTLSA/__arch/LICENSE b/...pandrel/architectures/DCTLSA/arch/LICENSE → ...ndrel/architectures/DCTLSA/__arch/LICENSE
diff --git a/...ndrel/architectures/DCTLSA/arch/dctlsa.py → ...rel/architectures/DCTLSA/__arch/dctlsa.py b/...ndrel/architectures/DCTLSA/arch/dctlsa.py → ...rel/architectures/DCTLSA/__arch/dctlsa.py
diff --git a/libs/spandrel/spandrel/architectures/DCTLSA/__init__.py b/libs/spandrel/spandrel/architectures/DCTLSA/__init__.py
@@ -8,7 +8,7 @@
     SizeRequirements,
     StateDict,
 )
-from .arch.dctlsa import DCTLSA
+from .__arch.dctlsa import DCTLSA
 
 
 class DCTLSAArch(Architecture[DCTLSA]):

diff --git a/...drel/architectures/DITN/arch/DITN_Real.py → ...el/architectures/DITN/__arch/DITN_Real.py b/...drel/architectures/DITN/arch/DITN_Real.py → ...el/architectures/DITN/__arch/DITN_Real.py
diff --git a/.../spandrel/architectures/DITN/arch/LICENSE → ...pandrel/architectures/DITN/__arch/LICENSE b/.../spandrel/architectures/DITN/arch/LICENSE → ...pandrel/architectures/DITN/__arch/LICENSE
diff --git a/libs/spandrel/spandrel/architectures/DITN/__init__.py b/libs/spandrel/spandrel/architectures/DITN/__init__.py
@@ -12,7 +12,7 @@
     SizeRequirements,
     StateDict,
 )
-from .arch.DITN_Real import DITN_Real as DITN
+from .__arch.DITN_Real import DITN_Real as DITN
 
 
 class DITNArch(Architecture[DITN]):

diff --git a/.../spandrel/architectures/DRCT/arch/LICENSE → ...pandrel/architectures/DRCT/__arch/LICENSE b/.../spandrel/architectures/DRCT/arch/LICENSE → ...pandrel/architectures/DRCT/__arch/LICENSE
diff --git a/...drel/architectures/DRCT/arch/drct_arch.py → ...el/architectures/DRCT/__arch/drct_arch.py b/...drel/architectures/DRCT/arch/drct_arch.py → ...el/architectures/DRCT/__arch/drct_arch.py
diff --git a/libs/spandrel/spandrel/architectures/DRCT/__init__.py b/libs/spandrel/spandrel/architectures/DRCT/__init__.py
@@ -10,7 +10,7 @@
     SizeRequirements,
     StateDict,
 )
-from .arch.drct_arch import DRCT
+from .__arch.drct_arch import DRCT
 
 
 class DRCTArch(Architecture[DRCT]):

diff --git a/...pandrel/architectures/DRUNet/arch/LICENSE → ...ndrel/architectures/DRUNet/__arch/LICENSE b/...pandrel/architectures/DRUNet/arch/LICENSE → ...ndrel/architectures/DRUNet/__arch/LICENSE
diff --git a/...architectures/DRUNet/arch/network_unet.py → ...chitectures/DRUNet/__arch/network_unet.py b/...architectures/DRUNet/arch/network_unet.py → ...chitectures/DRUNet/__arch/network_unet.py
diff --git a/libs/spandrel/spandrel/architectures/DRUNet/__init__.py b/libs/spandrel/spandrel/architectures/DRUNet/__init__.py
@@ -11,7 +11,7 @@
     SizeRequirements,
     StateDict,
 )
-from .arch.network_unet import DRUNet
+from .__arch.network_unet import DRUNet
 
 
 class DRUNetArch(Architecture[DRUNet]):

diff --git a/...spandrel/architectures/DnCNN/arch/LICENSE → ...andrel/architectures/DnCNN/__arch/LICENSE b/...spandrel/architectures/DnCNN/arch/LICENSE → ...andrel/architectures/DnCNN/__arch/LICENSE
diff --git a/...architectures/DnCNN/arch/network_dncnn.py → ...chitectures/DnCNN/__arch/network_dncnn.py b/...architectures/DnCNN/arch/network_dncnn.py → ...chitectures/DnCNN/__arch/network_dncnn.py
diff --git a/libs/spandrel/spandrel/architectures/DnCNN/__init__.py b/libs/spandrel/spandrel/architectures/DnCNN/__init__.py
@@ -13,7 +13,7 @@
     SizeRequirements,
     StateDict,
 )
-from .arch.network_dncnn import DnCNN
+from .__arch.network_dncnn import DnCNN
 
 
 class DnCNNArch(Architecture[DnCNN]):

diff --git a/...pandrel/architectures/ESRGAN/arch/LICENSE → ...ndrel/architectures/ESRGAN/__arch/LICENSE b/...pandrel/architectures/ESRGAN/arch/LICENSE → ...ndrel/architectures/ESRGAN/__arch/LICENSE
diff --git a/...pandrel/architectures/ESRGAN/arch/RRDB.py → ...ndrel/architectures/ESRGAN/__arch/RRDB.py b/...pandrel/architectures/ESRGAN/arch/RRDB.py → ...ndrel/architectures/ESRGAN/__arch/RRDB.py
diff --git a/libs/spandrel/spandrel/architectures/ESRGAN/__init__.py b/libs/spandrel/spandrel/architectures/ESRGAN/__init__.py
@@ -15,7 +15,7 @@
     SizeRequirements,
     StateDict,
 )
-from .arch.RRDB import RRDBNet
+from .__arch.RRDB import RRDBNet
 
 
 def _new_to_old_arch(state: StateDict, state_map: dict, num_blocks: int):

diff --git a/...pandrel/architectures/FBCNN/arch/FBCNN.py → ...ndrel/architectures/FBCNN/__arch/FBCNN.py b/...pandrel/architectures/FBCNN/arch/FBCNN.py → ...ndrel/architectures/FBCNN/__arch/FBCNN.py
diff --git a/...spandrel/architectures/FBCNN/arch/LICENSE → ...andrel/architectures/FBCNN/__arch/LICENSE b/...spandrel/architectures/FBCNN/arch/LICENSE → ...andrel/architectures/FBCNN/__arch/LICENSE
diff --git a/libs/spandrel/spandrel/architectures/FBCNN/__init__.py b/libs/spandrel/spandrel/architectures/FBCNN/__init__.py
@@ -3,7 +3,7 @@
 from spandrel.util import KeyCondition, get_seq_len
 
 from ...__helpers.model_descriptor import Architecture, ImageModelDescriptor, StateDict
-from .arch.FBCNN import FBCNN
+from .__arch.FBCNN import FBCNN
 
 
 class FBCNNArch(Architecture[FBCNN]):

diff --git a/...drel/architectures/FFTformer/arch/LICENSE → ...el/architectures/FFTformer/__arch/LICENSE b/...drel/architectures/FFTformer/arch/LICENSE → ...el/architectures/FFTformer/__arch/LICENSE
diff --git a/...tectures/FFTformer/arch/fftformer_arch.py → ...ctures/FFTformer/__arch/fftformer_arch.py b/...tectures/FFTformer/arch/fftformer_arch.py → ...ctures/FFTformer/__arch/fftformer_arch.py
diff --git a/libs/spandrel/spandrel/architectures/FFTformer/__init__.py b/libs/spandrel/spandrel/architectures/FFTformer/__init__.py
@@ -10,7 +10,7 @@
     SizeRequirements,
     StateDict,
 )
-from .arch.fftformer_arch import FFTformer
+from .__arch.fftformer_arch import FFTformer
 
 
 class FFTformerArch(Architecture[FFTformer]):

diff --git a/...pandrel/architectures/GFPGAN/arch/LICENSE → ...ndrel/architectures/GFPGAN/__arch/LICENSE b/...pandrel/architectures/GFPGAN/arch/LICENSE → ...ndrel/architectures/GFPGAN/__arch/LICENSE
diff --git a/...architectures/GFPGAN/arch/arcface_arch.py → ...chitectures/GFPGAN/__arch/arcface_arch.py b/...architectures/GFPGAN/arch/arcface_arch.py → ...chitectures/GFPGAN/__arch/arcface_arch.py
diff --git a/...el/architectures/GFPGAN/arch/fused_act.py → .../architectures/GFPGAN/__arch/fused_act.py b/...el/architectures/GFPGAN/arch/fused_act.py → .../architectures/GFPGAN/__arch/fused_act.py
diff --git a/...tures/GFPGAN/arch/gfpgan_bilinear_arch.py → ...res/GFPGAN/__arch/gfpgan_bilinear_arch.py b/...tures/GFPGAN/arch/gfpgan_bilinear_arch.py → ...res/GFPGAN/__arch/gfpgan_bilinear_arch.py
diff --git a/...rchitectures/GFPGAN/arch/gfpganv1_arch.py → ...hitectures/GFPGAN/__arch/gfpganv1_arch.py b/...rchitectures/GFPGAN/arch/gfpganv1_arch.py → ...hitectures/GFPGAN/__arch/gfpganv1_arch.py
diff --git a/...ctures/GFPGAN/arch/gfpganv1_clean_arch.py → ...ures/GFPGAN/__arch/gfpganv1_clean_arch.py b/...ctures/GFPGAN/arch/gfpganv1_clean_arch.py → ...ures/GFPGAN/__arch/gfpganv1_clean_arch.py
diff --git a/...chitectures/GFPGAN/arch/stylegan2_arch.py → ...itectures/GFPGAN/__arch/stylegan2_arch.py b/...chitectures/GFPGAN/arch/stylegan2_arch.py → ...itectures/GFPGAN/__arch/stylegan2_arch.py
diff --git a/...es/GFPGAN/arch/stylegan2_bilinear_arch.py → .../GFPGAN/__arch/stylegan2_bilinear_arch.py b/...es/GFPGAN/arch/stylegan2_bilinear_arch.py → .../GFPGAN/__arch/stylegan2_bilinear_arch.py
diff --git a/...tures/GFPGAN/arch/stylegan2_clean_arch.py → ...res/GFPGAN/__arch/stylegan2_clean_arch.py b/...tures/GFPGAN/arch/stylegan2_clean_arch.py → ...res/GFPGAN/__arch/stylegan2_clean_arch.py
diff --git a/...el/architectures/GFPGAN/arch/upfirdn2d.py → .../architectures/GFPGAN/__arch/upfirdn2d.py b/...el/architectures/GFPGAN/arch/upfirdn2d.py → .../architectures/GFPGAN/__arch/upfirdn2d.py
diff --git a/libs/spandrel/spandrel/architectures/GFPGAN/__init__.py b/libs/spandrel/spandrel/architectures/GFPGAN/__init__.py
@@ -8,7 +8,7 @@
     SizeRequirements,
     StateDict,
 )
-from .arch.gfpganv1_clean_arch import GFPGANv1Clean
+from .__arch.gfpganv1_clean_arch import GFPGANv1Clean
 
 
 class GFPGANArch(Architecture[GFPGANv1Clean]):

diff --git a/...l/spandrel/architectures/GRL/arch/LICENSE → ...spandrel/architectures/GRL/__arch/LICENSE b/...l/spandrel/architectures/GRL/arch/LICENSE → ...spandrel/architectures/GRL/__arch/LICENSE
diff --git a/...spandrel/architectures/GRL/arch/config.py → ...andrel/architectures/GRL/__arch/config.py b/...spandrel/architectures/GRL/arch/config.py → ...andrel/architectures/GRL/__arch/config.py
diff --git a/...el/spandrel/architectures/GRL/arch/grl.py → .../spandrel/architectures/GRL/__arch/grl.py b/...el/spandrel/architectures/GRL/arch/grl.py → .../spandrel/architectures/GRL/__arch/grl.py
diff --git a/...rchitectures/GRL/arch/mixed_attn_block.py → ...hitectures/GRL/__arch/mixed_attn_block.py b/...rchitectures/GRL/arch/mixed_attn_block.py → ...hitectures/GRL/__arch/mixed_attn_block.py
diff --git a/...es/GRL/arch/mixed_attn_block_efficient.py → .../GRL/__arch/mixed_attn_block_efficient.py b/...es/GRL/arch/mixed_attn_block_efficient.py → .../GRL/__arch/mixed_attn_block_efficient.py
diff --git a/...el/spandrel/architectures/GRL/arch/ops.py → .../spandrel/architectures/GRL/__arch/ops.py b/...el/spandrel/architectures/GRL/arch/ops.py → .../spandrel/architectures/GRL/__arch/ops.py
diff --git a/...l/architectures/GRL/arch/swin_v1_block.py → ...architectures/GRL/__arch/swin_v1_block.py b/...l/architectures/GRL/arch/swin_v1_block.py → ...architectures/GRL/__arch/swin_v1_block.py
diff --git a/...andrel/architectures/GRL/arch/upsample.py → ...drel/architectures/GRL/__arch/upsample.py b/...andrel/architectures/GRL/arch/upsample.py → ...drel/architectures/GRL/__arch/upsample.py
diff --git a/libs/spandrel/spandrel/architectures/GRL/__init__.py b/libs/spandrel/spandrel/architectures/GRL/__init__.py
@@ -15,7 +15,7 @@
 
 from ...__helpers.canonicalize import remove_common_prefix
 from ...__helpers.model_descriptor import Architecture, ImageModelDescriptor, StateDict
-from .arch.grl import GRL
+from .__arch.grl import GRL
 
 _NON_PERSISTENT_BUFFERS = [
     "table_w",

diff --git a/...el/spandrel/architectures/HAT/arch/HAT.py → .../spandrel/architectures/HAT/__arch/HAT.py b/...el/spandrel/architectures/HAT/arch/HAT.py → .../spandrel/architectures/HAT/__arch/HAT.py
diff --git a/...l/spandrel/architectures/HAT/arch/LICENSE → ...spandrel/architectures/HAT/__arch/LICENSE b/...l/spandrel/architectures/HAT/arch/LICENSE → ...spandrel/architectures/HAT/__arch/LICENSE
diff --git a/libs/spandrel/spandrel/architectures/HAT/__init__.py b/libs/spandrel/spandrel/architectures/HAT/__init__.py
@@ -10,7 +10,7 @@
     SizeRequirements,
     StateDict,
 )
-from .arch.HAT import HAT
+from .__arch.HAT import HAT
 
 
 def _get_overlap_ratio(window_size: int, with_overlap: int) -> float:

diff --git a/...itectures/HVICIDNet/arch/HVI_transform.py → ...ectures/HVICIDNet/__arch/HVI_transform.py b/...itectures/HVICIDNet/arch/HVI_transform.py → ...ectures/HVICIDNet/__arch/HVI_transform.py
diff --git a/...ndrel/architectures/HVICIDNet/arch/LCA.py → ...rel/architectures/HVICIDNet/__arch/LCA.py b/...ndrel/architectures/HVICIDNet/arch/LCA.py → ...rel/architectures/HVICIDNet/__arch/LCA.py
diff --git a/...drel/architectures/HVICIDNet/arch/LICENSE → ...el/architectures/HVICIDNet/__arch/LICENSE b/...drel/architectures/HVICIDNet/arch/LICENSE → ...el/architectures/HVICIDNet/__arch/LICENSE
diff --git a/...el/architectures/HVICIDNet/arch/cidnet.py → .../architectures/HVICIDNet/__arch/cidnet.py b/...el/architectures/HVICIDNet/arch/cidnet.py → .../architectures/HVICIDNet/__arch/cidnet.py
diff --git a/...tures/HVICIDNet/arch/transformer_utils.py → ...res/HVICIDNet/__arch/transformer_utils.py b/...tures/HVICIDNet/arch/transformer_utils.py → ...res/HVICIDNet/__arch/transformer_utils.py
diff --git a/libs/spandrel/spandrel/architectures/HVICIDNet/__init__.py b/libs/spandrel/spandrel/architectures/HVICIDNet/__init__.py
@@ -9,7 +9,7 @@
     SizeRequirements,
     StateDict,
 )
-from .arch.cidnet import CIDNet as HVICIDNet
+from .__arch.cidnet import CIDNet as HVICIDNet
 
 
 class HVICIDNetArch(Architecture[HVICIDNet]):

diff --git a/...l/spandrel/architectures/IPT/arch/LICENSE → ...spandrel/architectures/IPT/__arch/LICENSE b/...l/spandrel/architectures/IPT/arch/LICENSE → ...spandrel/architectures/IPT/__arch/LICENSE
diff --git a/...spandrel/architectures/IPT/arch/common.py → ...andrel/architectures/IPT/__arch/common.py b/...spandrel/architectures/IPT/arch/common.py → ...andrel/architectures/IPT/__arch/common.py
diff --git a/...el/spandrel/architectures/IPT/arch/ipt.py → .../spandrel/architectures/IPT/__arch/ipt.py b/...el/spandrel/architectures/IPT/arch/ipt.py → .../spandrel/architectures/IPT/__arch/ipt.py
diff --git a/.../spandrel/architectures/IPT/arch/model.py → ...pandrel/architectures/IPT/__arch/model.py b/.../spandrel/architectures/IPT/arch/model.py → ...pandrel/architectures/IPT/__arch/model.py
diff --git a/libs/spandrel/spandrel/architectures/IPT/__init__.py b/libs/spandrel/spandrel/architectures/IPT/__init__.py
@@ -14,8 +14,8 @@
     SizeRequirements,
     StateDict,
 )
-from .arch.ipt import IPT
-from .arch.model import Model
+from .__arch.ipt import IPT
+from .__arch.model import Model
 
 
 class IPTArch(Architecture[IPT]):

diff --git a/...spandrel/architectures/KBNet/arch/LICENSE → ...andrel/architectures/KBNet/__arch/LICENSE b/...spandrel/architectures/KBNet/arch/LICENSE → ...andrel/architectures/KBNet/__arch/LICENSE
diff --git a/...drel/architectures/KBNet/arch/kb_utils.py → ...el/architectures/KBNet/__arch/kb_utils.py b/...drel/architectures/KBNet/arch/kb_utils.py → ...el/architectures/KBNet/__arch/kb_utils.py
diff --git a/...ndrel/architectures/KBNet/arch/kbnet_l.py → ...rel/architectures/KBNet/__arch/kbnet_l.py b/...ndrel/architectures/KBNet/arch/kbnet_l.py → ...rel/architectures/KBNet/__arch/kbnet_l.py
diff --git a/...ndrel/architectures/KBNet/arch/kbnet_s.py → ...rel/architectures/KBNet/__arch/kbnet_s.py b/...ndrel/architectures/KBNet/arch/kbnet_s.py → ...rel/architectures/KBNet/__arch/kbnet_s.py
diff --git a/libs/spandrel/spandrel/architectures/KBNet/__init__.py b/libs/spandrel/spandrel/architectures/KBNet/__init__.py
@@ -12,8 +12,8 @@
     SizeRequirements,
     StateDict,
 )
-from .arch.kbnet_l import KBNet_l
-from .arch.kbnet_s import KBNet_s
+from .__arch.kbnet_l import KBNet_l
+from .__arch.kbnet_s import KBNet_s
 
 # KBCNN is essentially 2 similar but different architectures: KBNet_l and KBNet_s.
 _KBNet = Union[KBNet_l, KBNet_s]

diff --git a/.../spandrel/architectures/LaMa/arch/LICENSE → ...pandrel/architectures/LaMa/__arch/LICENSE b/.../spandrel/architectures/LaMa/arch/LICENSE → ...pandrel/architectures/LaMa/__arch/LICENSE
diff --git a/.../spandrel/architectures/LaMa/arch/LaMa.py → ...pandrel/architectures/LaMa/__arch/LaMa.py b/.../spandrel/architectures/LaMa/arch/LaMa.py → ...pandrel/architectures/LaMa/__arch/LaMa.py
diff --git a/libs/spandrel/spandrel/architectures/LaMa/__init__.py b/libs/spandrel/spandrel/architectures/LaMa/__init__.py
@@ -8,7 +8,7 @@
     SizeRequirements,
     StateDict,
 )
-from .arch.LaMa import LaMa
+from .__arch.LaMa import LaMa
 
 
 class LaMaArch(Architecture[LaMa]):

diff --git a/...ndrel/architectures/MMRealSR/arch/LICENSE → ...rel/architectures/MMRealSR/__arch/LICENSE b/...ndrel/architectures/MMRealSR/arch/LICENSE → ...rel/architectures/MMRealSR/__arch/LICENSE
diff --git a/...hitectures/MMRealSR/arch/mmrealsr_arch.py → ...tectures/MMRealSR/__arch/mmrealsr_arch.py b/...hitectures/MMRealSR/arch/mmrealsr_arch.py → ...tectures/MMRealSR/__arch/mmrealsr_arch.py
diff --git a/libs/spandrel/spandrel/architectures/MMRealSR/__init__.py b/libs/spandrel/spandrel/architectures/MMRealSR/__init__.py
@@ -12,7 +12,7 @@
     SizeRequirements,
     StateDict,
 )
-from .arch.mmrealsr_arch import MMRRDBNet_test as MMRealSR
+from .__arch.mmrealsr_arch import MMRRDBNet_test as MMRealSR
 
 
 def _get_in_ch_and_scale(combined: int, num_out_ch: int) -> tuple[int, int]:

diff --git a/...l/architectures/MixDehazeNet/arch/LICENSE → ...architectures/MixDehazeNet/__arch/LICENSE b/...l/architectures/MixDehazeNet/arch/LICENSE → ...architectures/MixDehazeNet/__arch/LICENSE
diff --git a/...ectures/MixDehazeNet/arch/MixDehazeNet.py → ...tures/MixDehazeNet/__arch/MixDehazeNet.py b/...ectures/MixDehazeNet/arch/MixDehazeNet.py → ...tures/MixDehazeNet/__arch/MixDehazeNet.py
diff --git a/libs/spandrel/spandrel/architectures/MixDehazeNet/__init__.py b/libs/spandrel/spandrel/architectures/MixDehazeNet/__init__.py
@@ -11,7 +11,7 @@
     SizeRequirements,
     StateDict,
 )
-from .arch.MixDehazeNet import MixDehazeNet
+from .__arch.MixDehazeNet import MixDehazeNet
 
 
 class MixDehazeNetArch(Architecture[MixDehazeNet]):

diff --git a/...pandrel/architectures/NAFNet/arch/LICENSE → ...ndrel/architectures/NAFNet/__arch/LICENSE b/...pandrel/architectures/NAFNet/arch/LICENSE → ...ndrel/architectures/NAFNet/__arch/LICENSE
diff --git a/.../architectures/NAFNet/arch/NAFNet_arch.py → ...rchitectures/NAFNet/__arch/NAFNet_arch.py b/.../architectures/NAFNet/arch/NAFNet_arch.py → ...rchitectures/NAFNet/__arch/NAFNet_arch.py
diff --git a/...el/architectures/NAFNet/arch/arch_util.py → .../architectures/NAFNet/__arch/arch_util.py b/...el/architectures/NAFNet/arch/arch_util.py → .../architectures/NAFNet/__arch/arch_util.py
diff --git a/libs/spandrel/spandrel/architectures/NAFNet/__init__.py b/libs/spandrel/spandrel/architectures/NAFNet/__init__.py
@@ -5,7 +5,7 @@
 from spandrel.util import KeyCondition, get_seq_len
 
 from ...__helpers.model_descriptor import Architecture, ImageModelDescriptor, StateDict
-from .arch.NAFNet_arch import NAFNet
+from .__arch.NAFNet_arch import NAFNet
 
 
 class NAFNetArch(Architecture[NAFNet]):

diff --git a/...itectures/OmniSR/arch/ChannelAttention.py → ...ectures/OmniSR/__arch/ChannelAttention.py b/...itectures/OmniSR/arch/ChannelAttention.py → ...ectures/OmniSR/__arch/ChannelAttention.py
diff --git a/...pandrel/architectures/OmniSR/arch/LICENSE → ...ndrel/architectures/OmniSR/__arch/LICENSE b/...pandrel/architectures/OmniSR/arch/LICENSE → ...ndrel/architectures/OmniSR/__arch/LICENSE
diff --git a/...spandrel/architectures/OmniSR/arch/OSA.py → ...andrel/architectures/OmniSR/__arch/OSA.py b/...spandrel/architectures/OmniSR/arch/OSA.py → ...andrel/architectures/OmniSR/__arch/OSA.py
diff --git a/...pandrel/architectures/OmniSR/arch/OSAG.py → ...ndrel/architectures/OmniSR/__arch/OSAG.py b/...pandrel/architectures/OmniSR/arch/OSAG.py → ...ndrel/architectures/OmniSR/__arch/OSAG.py
diff --git a/...ndrel/architectures/OmniSR/arch/OmniSR.py → ...rel/architectures/OmniSR/__arch/OmniSR.py b/...ndrel/architectures/OmniSR/arch/OmniSR.py → ...rel/architectures/OmniSR/__arch/OmniSR.py
diff --git a/...spandrel/architectures/OmniSR/arch/esa.py → ...andrel/architectures/OmniSR/__arch/esa.py b/...spandrel/architectures/OmniSR/arch/esa.py → ...andrel/architectures/OmniSR/__arch/esa.py
diff --git a/...el/architectures/OmniSR/arch/layernorm.py → .../architectures/OmniSR/__arch/layernorm.py b/...el/architectures/OmniSR/arch/layernorm.py → .../architectures/OmniSR/__arch/layernorm.py
diff --git a/...architectures/OmniSR/arch/pixelshuffle.py → ...chitectures/OmniSR/__arch/pixelshuffle.py b/...architectures/OmniSR/arch/pixelshuffle.py → ...chitectures/OmniSR/__arch/pixelshuffle.py
diff --git a/libs/spandrel/spandrel/architectures/OmniSR/__init__.py b/libs/spandrel/spandrel/architectures/OmniSR/__init__.py
@@ -10,7 +10,7 @@
     SizeRequirements,
     StateDict,
 )
-from .arch.OmniSR import OmniSR
+from .__arch.OmniSR import OmniSR
 
 
 class OmniSRArch(Architecture[OmniSR]):

diff --git a/...spandrel/architectures/PLKSR/arch/LICENSE → ...andrel/architectures/PLKSR/__arch/LICENSE b/...spandrel/architectures/PLKSR/arch/LICENSE → ...andrel/architectures/PLKSR/__arch/LICENSE
@@ -1,21 +1,21 @@
-MIT License
-
-Copyright (c) 2024 Dongheon Lee
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in all
-copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.
+MIT License
+
+Copyright (c) 2024 Dongheon Lee
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.
Original file line number	Diff line number	Diff line change
Expand Up		@@ -169,4 +169,4 @@ Here are some cool projects that use Spandrel:

		## License

		This repo is bounded by the MIT license. However, the code of implemented architectures (everything inside an `arch/` directory) is bound by their original respective licenses (which are included in their respective `arch/` directories).
		This repo is bounded by the MIT license. However, the code of implemented architectures (everything inside an `__arch/` directory) is bound by their original respective licenses (which are included in their respective `__arch/` directories).