diff --git a/modules/infotext.py b/modules/infotext.py index 75a955e2d..407395c1e 100644 --- a/modules/infotext.py +++ b/modules/infotext.py @@ -96,6 +96,7 @@ mapping = [ # Other ('Parser', 'prompt_attention'), ('Color correction', 'img2img_color_correction'), + ('Color correction method', 'color_correction_method'), # Samplers ('Sampler eta delta', 'eta_noise_seed_delta'), ('Sampler eta multiplier', 'initial_noise_multiplier'),