aljaz/InvSR
1
0
Fork 0
mirror of https://github.com/aljazceru/InvSR.git synced 2025-12-17 06:14:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

add chopping size options

Browse Source
This commit is contained in:
zsyOAOA
2024-12-11 21:00:33 +08:00
parent 8d7e434e4f
commit f980a9df14
2 changed files with 14 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
sampler_invsr.py
View File

@@ -134,8 +134,6 @@ class InvSamplerSR(BaseSampler):
diffusion_sf = self.sd_pipe.transformer.patch_size
mod_lq = vae_sf // self.configs.basesr.sf * diffusion_sf
idle_pch_size = self.configs.basesr.chopping.pch_size
if ori_h_lq * ori_w_lq >= 512 ** 2:
idle_pch_size = 256
if min(im_cond.shape[-2:]) >= idle_pch_size:
pad_h_up = pad_w_left = 0