video implement sampler shift

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-10-17 14:43:41 -04:00
parent 0fc9072bf3
commit 9389aa710a
3 changed files with 14 additions and 1 deletions
+1
View File
@@ -7,6 +7,7 @@ Post-release fixes:
- SDNQ-SVD fallback on incompatible layers
- Huggingface model download
- Remove unused UI settings
- Video implement dynamic and manual sampler shift
## Update for 2025-10-17
+1 -1
View File
@@ -51,7 +51,7 @@ def find_sampler_config(name):
def restore_default(model):
if model is None:
return None
if getattr(model, "default_scheduler", None) is not None:
if getattr(model, "default_scheduler", None) is not None and getattr(model, "scheduler") is not None:
model.scheduler = copy.deepcopy(model.default_scheduler)
if hasattr(model, "prior_pipe") and hasattr(model.prior_pipe, "scheduler"):
model.prior_pipe.scheduler = copy.deepcopy(model.default_scheduler)
+12
View File
@@ -1,4 +1,5 @@
import os
import copy
import time
from modules import shared, errors, sd_models, processing, devices, images, ui_common
from modules.video_models import models_def, video_utils, video_load, video_vae, video_overrides, video_save, video_prompt
@@ -92,10 +93,21 @@ def generate(*args, **kwargs):
p.task_args['height'] = p.height
p.task_args['output_type'] = 'latent' if (p.vae_type == 'Remote') else 'pil'
p.ops.append('video')
# set scheduler params
orig_dynamic_shift = shared.opts.schedulers_dynamic_shift
orig_sampler_shift = shared.opts.schedulers_shift
shared.opts.data['schedulers_dynamic_shift'] = dynamic_shift
shared.opts.data['schedulers_shift'] = sampler_shift
if hasattr(shared.sd_model.scheduler, 'config') and hasattr(shared.sd_model.scheduler, 'register_to_config'):
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'):
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)
video_overrides.set_overrides(p, selected)
debug(f'Video: task_args={p.task_args}')