nodes/kijai-ComfyUI-SUPIR
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix tiled sampling

Browse Source
This commit is contained in:
kijai
2024-03-11 09:31:27 +02:00
parent aadedb1d29
commit 8a4beaa547
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+3 -1
View File
@@ -229,9 +229,11 @@ class SUPIR_Upscale:
config = OmegaConf.load(config_path_tiled)
config.model.params.sampler_config.params.tile_size = sampler_tile_size // 8
config.model.params.sampler_config.params.tile_stride = sampler_tile_stride // 8
config.model.params.sampler_config.target = f".sgm.modules.diffusionmodules.sampling.Tiled{sampler}"
print("Using tiled sampling")
else:
config = OmegaConf.load(config_path)
config.model.params.sampler_config.target = f".sgm.modules.diffusionmodules.sampling.{sampler}"
print("Using non-tiled sampling")
if XFORMERS_IS_AVAILABLE:
@@ -241,7 +243,7 @@ class SUPIR_Upscale:
config.model.params.ae_dtype = vae_dtype
config.model.params.diffusion_dtype = model_dtype
config.model.params.sampler_config.target = f".sgm.modules.diffusionmodules.sampling.{sampler}"
self.model = instantiate_from_config(config.model).cpu()
try: