Skip to content

Support Custom ESRGAN tile size#896

Merged
leejet merged 5 commits into
leejet:masterfrom
pedroCabrera:esrgan-tile-size
Dec 10, 2025
Merged

Support Custom ESRGAN tile size#896
leejet merged 5 commits into
leejet:masterfrom
pedroCabrera:esrgan-tile-size

Conversation

@pedroCabrera

Copy link
Copy Markdown
Contributor

Add suport for custom tile size in esrgan models ( this allows for faster inference on graphics card with good amount of memory )

@wbruna

wbruna commented Oct 18, 2025

Copy link
Copy Markdown
Contributor

Working fine on a few tests. Increasing the tile size didn't really help my card, even though it has 16G VRAM, but lowering it to 64 made it almost twice as fast.

@pedroCabrera , the recent command-line flag refactoring caused conflicts; could you take a look?

I'd also suggest changing the command-line flag from --upscale-tile to --upscale-tile-size, to better match --vae-tile-size. That would also make it a bit more future-proof, in case we add support for changing the overlapping factor, non-square tiles, etc.

@pedroCabrera

Copy link
Copy Markdown
Contributor Author

Thanks for checking, I have updated the param to --upscale-tile-size and rebased with latest changes so it should be mergeable i think!

@leejet leejet merged commit 1ac5a61 into leejet:master Dec 10, 2025
9 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants