mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 09:14:35 +02:00
video implement sampler shift
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
@@ -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)
|
||||
|
||||
@@ -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}')
|
||||
|
||||
|
||||
Reference in New Issue
Block a user