diff --git a/CHANGELOG.md b/CHANGELOG.md index 15f336018..41ea895de 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -15,6 +15,7 @@ - `processors` init code and multiple fixes - `live preview` configurable pause when not in focus, thanks @Artheriax - `mps` install `torchsde` as requirement + - `vae` scale factor improved detection ## Update for 2026-06-16 diff --git a/modules/control/run.py b/modules/control/run.py index 086c7eac5..4c9834aa5 100644 --- a/modules/control/run.py +++ b/modules/control/run.py @@ -356,7 +356,7 @@ def control_run(state: str = '', # pylint: disable=keyword-arg-before-vararg grading_shadows_tint: str = "#000000", grading_highlights_tint: str = "#ffffff", grading_split_tone_balance: float = 0.5, grading_vignette: float = 0.0, grading_grain: float = 0.0, grading_lut_file: str = "", grading_lut_strength: float = 1.0, - resize_mode_before: int = 0, resize_name_before: str = 'None', resize_context_before: str = 'None', width_before: int = 512, height_before: int = 512, scale_by_before: float = 1.0, selected_scale_tab_before: int = 0, + resize_mode_before: int = 0, resize_name_before: str = 'None', resize_context_before: str = 'None', width_before: int = 1024, height_before: int = 1024, scale_by_before: float = 1.0, selected_scale_tab_before: int = 0, resize_mode_after: int = 0, resize_name_after: str = 'None', resize_context_after: str = 'None', width_after: int = 0, height_after: int = 0, scale_by_after: float = 1.0, selected_scale_tab_after: int = 0, resize_mode_mask: int = 0, resize_name_mask: str = 'None', resize_context_mask: str = 'None', width_mask: int = 0, height_mask: int = 0, scale_by_mask: float = 1.0, selected_scale_tab_mask: int = 0, denoising_strength: float = 0.3, diff --git a/modules/sd_vae.py b/modules/sd_vae.py index e03a01c34..ac436ad58 100644 --- a/modules/sd_vae.py +++ b/modules/sd_vae.py @@ -23,6 +23,7 @@ unspecified = object() vae_scale_override = { 'WanPipeline': 16, 'ChronoEditPipeline': 16, + 'AutoencoderKLWan': 16, } @@ -37,6 +38,8 @@ def get_vae_scale_factor(model: DiffusionPipeline | None = None): vae_scale_factor = 8 elif model.__class__.__name__ in vae_scale_override: vae_scale_factor = vae_scale_override[model.__class__.__name__] + elif hasattr(model, 'vae') and model.vae.__class__.__name__ in vae_scale_override: + vae_scale_factor = vae_scale_override[model.vae.__class__.__name__] elif hasattr(model, 'vae_scale_factor_spatial'): vae_scale_factor = model.vae_scale_factor_spatial elif hasattr(model, 'vae_scale_factor'):