Skip to content

Commit

Permalink
Merge pull request #1141 from bghira/feature/vae-tiled-encode
Browse files Browse the repository at this point in the history
add --vae_enable_tiling to encode large res images with less vram used
  • Loading branch information
bghira authored Nov 11, 2024
2 parents cbebba4 + 5927f57 commit c9711f7
Showing 1 changed file with 9 additions and 0 deletions.
9 changes: 9 additions & 0 deletions helpers/configuration/cmd_args.py
Original file line number Diff line number Diff line change
Expand Up @@ -592,6 +592,15 @@ def get_argument_parser():
" but if you are at that point of contention, it's possible that your GPU has too little RAM. Default: 4."
),
)
parser.add_argument(
"--vae_enable_tiling",
action="store_true",
default=False,
help=(
"If set, will enable tiling for VAE caching. This is useful for very large images when VRAM is limited."
" This may be required for 2048px VAE caching on 24G accelerators, in addition to reducing --vae_batch_size."
),
)
parser.add_argument(
"--vae_cache_scan_behaviour",
type=str,
Expand Down

0 comments on commit c9711f7

Please sign in to comment.