mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 09:14:35 +02:00
fix wan22 stage selection
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
+3
-1
@@ -1,6 +1,6 @@
|
||||
# Change Log for SD.Next
|
||||
|
||||
## Update for 2025-10-19
|
||||
## Update for 2025-10-21
|
||||
|
||||
- **Features**
|
||||
- **offline mode**: enable in *settings -> hugginface*
|
||||
@@ -16,6 +16,8 @@
|
||||
- **Fixes**
|
||||
- startup error with `--profile` enabled if using `--skip`
|
||||
- restore orig init image for each batch sequence
|
||||
- fix modernui hints layout
|
||||
- fix `wan-2.2-a14b` stage selection
|
||||
|
||||
## Update for 2025-10-18
|
||||
|
||||
|
||||
Submodule extensions-builtin/sdnext-modernui updated: 8f6427aa03...80d1030c6b
@@ -19,14 +19,14 @@ def load_override(selected: Model):
|
||||
# WAN
|
||||
if 'WAN 2.1 14B' in selected.name:
|
||||
kwargs['vae'] = diffusers.AutoencoderKLWan.from_pretrained(selected.repo, subfolder="vae", torch_dtype=torch.float32, cache_dir=shared.opts.hfcache_dir)
|
||||
if 'A14B' in selected.name or '14B VACE' in selected.name:
|
||||
if ('A14B' in selected.name) or ('14B VACE' in selected.name):
|
||||
if shared.opts.model_wan_stage == 'combined':
|
||||
kwargs['boundary_ratio'] = shared.opts.model_wan_boundary
|
||||
elif shared.opts.model_wan_stage == 'high noise':
|
||||
kwargs['transformer_2'] = None
|
||||
kwargs['boundary_ratio'] = 0.0
|
||||
elif shared.opts.model_wan_stage == 'low noise':
|
||||
kwargs['boundary_ratio'] = 1.0
|
||||
kwargs['boundary_ratio'] = 1000.0
|
||||
kwargs['transformer'] = None
|
||||
debug(f'Video overrides: model="{selected.name}" kwargs={list(kwargs)}')
|
||||
return kwargs
|
||||
|
||||
@@ -103,7 +103,7 @@ def generate(*args, **kwargs):
|
||||
if hasattr(shared.sd_model.scheduler.config, 'use_dynamic_shifting'):
|
||||
shared.sd_model.scheduler.config.use_dynamic_shifting = dynamic_shift
|
||||
shared.sd_model.scheduler.register_to_config(use_dynamic_shifting = dynamic_shift)
|
||||
if hasattr(shared.sd_model.scheduler.config, 'flow_shift'):
|
||||
if hasattr(shared.sd_model.scheduler.config, 'flow_shift') and sampler_shift >= 0:
|
||||
shared.sd_model.scheduler.config.flow_shift = sampler_shift
|
||||
shared.sd_model.scheduler.register_to_config(flow_shift = sampler_shift)
|
||||
shared.sd_model.default_scheduler = copy.deepcopy(shared.sd_model.scheduler)
|
||||
|
||||
@@ -74,7 +74,7 @@ def load_wan(checkpoint_info, diffusers_load_config={}):
|
||||
elif shared.opts.model_wan_stage == 'low noise' or shared.opts.model_wan_stage == 'second':
|
||||
transformer = None
|
||||
transformer_2 = load_transformer(repo_id, diffusers_load_config, 'transformer_2')
|
||||
boundary_ratio = 1.0
|
||||
boundary_ratio = 1000.0
|
||||
elif shared.opts.model_wan_stage == 'combined' or shared.opts.model_wan_stage == 'both':
|
||||
transformer = load_transformer(repo_id, diffusers_load_config, 'transformer')
|
||||
transformer_2 = load_transformer(repo_id, diffusers_load_config, 'transformer_2')
|
||||
|
||||
Reference in New Issue
Block a user