add todo token downsampling and simplify tomes

This commit is contained in:
Vladimir Mandic
2024-04-28 20:25:33 -04:00
parent 08152cdc06
commit 9ff507d93a
16 changed files with 604 additions and 77 deletions
+2 -4
View File
@@ -131,10 +131,8 @@ def create_infotext(p: StableDiffusionProcessing, all_prompts=None, all_seeds=No
args['Sampler sigma noise'] = shared.opts.s_noise if shared.opts.s_noise != shared.opts.data_labels.get('s_noise').default else None
args['Sampler sigma tmin'] = shared.opts.s_tmin if shared.opts.s_tmin != shared.opts.data_labels.get('s_tmin').default else None
# tome
token_merging_ratio = p.get_token_merging_ratio()
token_merging_ratio_hr = p.get_token_merging_ratio(for_hr=True) if p.enable_hr else None
args['ToMe'] = token_merging_ratio if token_merging_ratio != 0 else None
args['ToMe hires'] = token_merging_ratio_hr if token_merging_ratio_hr != 0 else None
args['ToMe'] = shared.opts.tome_ratio if shared.opts.tome_ratio != 0 else None
args['ToDo'] = shared.opts.todo_ratio if shared.opts.todo_ratio != 0 else None
args.update(p.extra_generation_params)
params_text = ", ".join([k if k == v else f'{k}: {generation_parameters_copypaste.quote(v)}' for k, v in args.items() if v is not None])