jinx/ComfyUI
1
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-02 15:04:50 +08:00
Code Issues Activity

Tiled upscaling with the upscale models.

Browse Source
This commit is contained in:
comfyanonymous
2023-03-11 14:04:13 -05:00
parent 905857edd8
commit 9db2e97b47
3 changed files with 28 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
comfy_extras/nodes_upscale_model.py
View File

@@ -4,6 +4,7 @@ from comfy.sd import load_torch_file
import comfy.model_management
from nodes import filter_files_extensions, recursive_search, supported_ckpt_extensions
import torch
import comfy.utils
class UpscaleModelLoader:
models_dir = os.path.join(os.path.dirname(os.path.dirname(os.path.realpath(__file__))), "models")
@@ -40,8 +41,7 @@ class ImageUpscaleWithModel:
device = comfy.model_management.get_torch_device()
upscale_model.to(device)
in_img = image.movedim(-1,-3).to(device)
with torch.inference_mode():
s = upscale_model(in_img).cpu()
s = comfy.utils.tiled_scale(in_img, lambda a: upscale_model(a), tile_x=128 + 64, tile_y=128 + 64, overlap = 8, upscale_amount=upscale_model.scale)
upscale_model.cpu()
s = torch.clamp(s.movedim(-3,-1), min=0, max=1.0)
return (s,)