Skip to content

Conversation

pedroCabrera
Copy link
Contributor

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

@wbruna
Copy link
Contributor

wbruna commented Oct 18, 2025

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
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!

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