cleanup

Signed-off-by: Vladimir Mandic <mandic00@live.com>

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:

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)

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)