wehub-resource-sync
cddb07a176
build container image / cpu (push) Waiting to run
build container image / cuda (push) Waiting to run
build container image / rocm (push) Waiting to run
frontend tests / frontend-tests (push) Waiting to run
openapi checks / openapi-checks (push) Waiting to run
python tests / py3.12: macos-default (push) Waiting to run
python tests / py3.11: windows-cpu (push) Waiting to run
python tests / py3.12: windows-cpu (push) Waiting to run
python tests / py3.11: linux-cpu (push) Waiting to run
python tests / py3.12: linux-cpu (push) Waiting to run
typegen checks / typegen-checks (push) Waiting to run
uv lock checks / uv-lock-checks (push) Waiting to run
frontend checks / frontend-checks (push) Waiting to run
lfs checks / lfs-check (push) Waiting to run
python checks / python-checks (push) Waiting to run
python tests / py3.11: macos-default (push) Waiting to run
docs / deploy (push) Has been cancelled
docs / changes (push) Has been cancelled
docs / check-and-build (push) Has been cancelled
36 lines
1.3 KiB
Python
36 lines
1.3 KiB
Python
from contextlib import contextmanager
|
|
|
|
from diffusers.models.autoencoders.autoencoder_kl import AutoencoderKL
|
|
from diffusers.models.autoencoders.autoencoder_tiny import AutoencoderTiny
|
|
|
|
|
|
@contextmanager
|
|
def patch_vae_tiling_params(
|
|
vae: AutoencoderKL | AutoencoderTiny,
|
|
tile_sample_min_size: int,
|
|
tile_latent_min_size: int,
|
|
tile_overlap_factor: float,
|
|
):
|
|
"""Patch the parameters that control the VAE tiling tile size and overlap.
|
|
|
|
These parameters are not explicitly exposed in the VAE's API, but they have a significant impact on the quality of
|
|
the outputs. As a general rule, bigger tiles produce better results, but this comes at the cost of higher memory
|
|
usage.
|
|
"""
|
|
# Record initial config.
|
|
orig_tile_sample_min_size = vae.tile_sample_min_size
|
|
orig_tile_latent_min_size = vae.tile_latent_min_size
|
|
orig_tile_overlap_factor = vae.tile_overlap_factor
|
|
|
|
try:
|
|
# Apply target config.
|
|
vae.tile_sample_min_size = tile_sample_min_size
|
|
vae.tile_latent_min_size = tile_latent_min_size
|
|
vae.tile_overlap_factor = tile_overlap_factor
|
|
yield
|
|
finally:
|
|
# Restore initial config.
|
|
vae.tile_sample_min_size = orig_tile_sample_min_size
|
|
vae.tile_latent_min_size = orig_tile_latent_min_size
|
|
vae.tile_overlap_factor = orig_tile_overlap_factor
|