a  d[@sHddlZddlmZddlmZddlmZeGdddeZ dS)N) functional)MODEL_REGISTRY)SRModelc@seZdZddZdS) SwinIRModelc CsZ|jdd}|jdd}d\}}|j\}}}}||dkrN|||}||dkrf|||}t|jd|d|fd}t|dr|jt |||_ Wdn1s0YnH|j t | ||_ Wdn1s0Y|j |j \}}}}|j ddddd|||d|||f|_ dS) NZ network_g window_sizescaler)rrrZreflect net_g_ema)optgetZlqsizeFpadhasattrr evaltorchZno_gradoutputZnet_gZtrain) selfrrZ mod_pad_hZ mod_pad_w_hwimgrd/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/basicsr/models/swinir_model.pytest s&       ,  * zSwinIRModel.testN)__name__ __module__ __qualname__rrrrrrsr) rZtorch.nnrr Zbasicsr.utils.registryrZsr_modelrregisterrrrrrs