Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 09:14:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

vae_scale_factor override

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2026-06-20 10:49:43 +02:00
parent 8d71d5206d
commit afc1a043b6
3 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
CHANGELOG.md
+1
View File
@@ -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
modules/control/run.py
+1 -1
View File
@@ -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,
modules/sd_vae.py
+3
View File
@@ -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'):