diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-05-10 08:19:16 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-05-10 08:19:16 +0000 |
| commit | 550256db1ce18778a9d56ff343d844c61b9f9b83 (patch) | |
| tree | a17e8fd9cb475381c361844970ba2d9111938b6d /extensions-builtin/SwinIR/swinir_model_arch.py | |
| parent | 028d3f6425d85f122027c127fba8bcbf4f66ee75 (diff) | |
| download | stable-diffusion-webui-gfx803-550256db1ce18778a9d56ff343d844c61b9f9b83.tar.gz stable-diffusion-webui-gfx803-550256db1ce18778a9d56ff343d844c61b9f9b83.tar.bz2 stable-diffusion-webui-gfx803-550256db1ce18778a9d56ff343d844c61b9f9b83.zip | |
ruff manual fixes
Diffstat (limited to 'extensions-builtin/SwinIR/swinir_model_arch.py')
| -rw-r--r-- | extensions-builtin/SwinIR/swinir_model_arch.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/extensions-builtin/SwinIR/swinir_model_arch.py b/extensions-builtin/SwinIR/swinir_model_arch.py index 863f42db..75f7bedc 100644 --- a/extensions-builtin/SwinIR/swinir_model_arch.py +++ b/extensions-builtin/SwinIR/swinir_model_arch.py @@ -644,13 +644,17 @@ class SwinIR(nn.Module): """ def __init__(self, img_size=64, patch_size=1, in_chans=3, - embed_dim=96, depths=[6, 6, 6, 6], num_heads=[6, 6, 6, 6], + embed_dim=96, depths=None, num_heads=None, window_size=7, mlp_ratio=4., qkv_bias=True, qk_scale=None, drop_rate=0., attn_drop_rate=0., drop_path_rate=0.1, norm_layer=nn.LayerNorm, ape=False, patch_norm=True, use_checkpoint=False, upscale=2, img_range=1., upsampler='', resi_connection='1conv', **kwargs): super(SwinIR, self).__init__() + + depths = depths or [6, 6, 6, 6] + num_heads = num_heads or [6, 6, 6, 6] + num_in_ch = in_chans num_out_ch = in_chans num_feat = 64 |