From d4c8de99f59c90f2f1a7c13b3faaa815c9a276ca Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Sun, 23 Mar 2025 14:26:57 -0400 Subject: [PATCH] cleanup Signed-off-by: Vladimir Mandic --- modules/processing_args.py | 2 -- modules/sd_samplers.py | 5 +++-- modules/video_models/video_overrides.py | 1 - 3 files changed, 3 insertions(+), 5 deletions(-) diff --git a/modules/processing_args.py b/modules/processing_args.py index 058c8bdf1..db8794214 100644 --- a/modules/processing_args.py +++ b/modules/processing_args.py @@ -21,7 +21,6 @@ disable_pbar = os.environ.get('SD_DISABLE_PBAR', None) is not None def task_specific_kwargs(p, model): task_args = {} is_img2img_model = bool('Zero123' in shared.sd_model.__class__.__name__) - print('HERE', sd_models.get_diffusers_task(model)) if len(getattr(p, 'init_images', [])) > 0: if isinstance(p.init_images[0], str): p.init_images = [helpers.decode_base64_to_image(i, quiet=True) for i in p.init_images] @@ -35,7 +34,6 @@ def task_specific_kwargs(p, model): 'height': 8 * math.ceil(p.height / 8), } elif (sd_models.get_diffusers_task(model) == sd_models.DiffusersTaskType.IMAGE_2_IMAGE or is_img2img_model) and len(getattr(p, 'init_images', [])) > 0: - print('HERE1', p.denoising_strength) if shared.sd_model_type == 'sdxl' and hasattr(model, 'register_to_config'): model.register_to_config(requires_aesthetics_score = False) if 'hires' not in p.ops: diff --git a/modules/sd_samplers.py b/modules/sd_samplers.py index 67ed8e8ee..be375dbff 100644 --- a/modules/sd_samplers.py +++ b/modules/sd_samplers.py @@ -12,6 +12,8 @@ samplers = all_samplers samplers_for_img2img = all_samplers samplers_map = {} loaded_config = None +flow_models = ['Flux', 'StableDiffusion3', 'Lumina', 'AuraFlow', 'Sana', 'CogView4'] +flow_models += ['Hunyuan', 'LTX', 'Mochi'] def list_samplers(): @@ -79,10 +81,9 @@ def create_sampler(name, model): shared.log.debug(f'Sampler: "{name}" config={config.options}') return sampler elif shared.native: - FlowModels = ['Flux', 'StableDiffusion3', 'Lumina', 'AuraFlow', 'Sana', 'HunyuanVideoPipeline', 'CogView4Pipeline'] if 'KDiffusion' in model.__class__.__name__: return None - if not any(x in model.__class__.__name__ for x in FlowModels) and 'FlowMatch' in name: + if not any(x in model.__class__.__name__ for x in flow_models) and 'FlowMatch' in name: shared.log.warning(f'Sampler: default={current} target="{name}" class={model.__class__.__name__} flow-match scheduler unsupported') return None sampler = config.constructor(model) diff --git a/modules/video_models/video_overrides.py b/modules/video_models/video_overrides.py index e771f5254..7af00f6be 100644 --- a/modules/video_models/video_overrides.py +++ b/modules/video_models/video_overrides.py @@ -37,7 +37,6 @@ def set_overrides(p: processing.StableDiffusionProcessingVideo, selected: Model) if cls == 'LTXImageToVideoPipeline' or cls == 'LTXConditionPipeline': p.task_args['generator'] = None if cls == 'LTXConditionPipeline': - print('HERE2', p.denoising_strength) p.task_args['strength'] = p.denoising_strength if 'LTX' in shared.sd_model.__class__.__name__: p.task_args['width'] = 32 * (p.width // 32)