Commit 070b7d60 authored by Milly's avatar Milly Committed by AUTOMATIC1111
Browse files

Added styles to Processed

So `[styles]` pattern can use in saving image UI.
parent a5a08b0b
Loading
Loading
Loading
Loading
+1 −6
Original line number Diff line number Diff line
@@ -292,12 +292,7 @@ def apply_filename_pattern(x, p, seed, prompt):
        x = x.replace("[cfg]", str(p.cfg_scale))
        x = x.replace("[width]", str(p.width))
        x = x.replace("[height]", str(p.height))

        #currently disabled if using the save button, will work otherwise
        # if enabled it will cause a bug because styles is not included in the save_files data dictionary
        if hasattr(p, "styles"):
        x = x.replace("[styles]", sanitize_filename_part(", ".join([x for x in p.styles if not x == "None"]) or "None", replace_spaces=False))

        x = x.replace("[sampler]", sanitize_filename_part(sd_samplers.samplers[p.sampler_index].name, replace_spaces=False))

    x = x.replace("[model_hash]", shared.sd_model.sd_model_hash)
+2 −0
Original line number Diff line number Diff line
@@ -121,6 +121,7 @@ class Processed:
        self.denoising_strength = getattr(p, 'denoising_strength', None)
        self.extra_generation_params = p.extra_generation_params
        self.index_of_first_image = index_of_first_image
        self.styles = p.styles

        self.eta = p.eta
        self.ddim_discretize = p.ddim_discretize
@@ -165,6 +166,7 @@ class Processed:
            "extra_generation_params": self.extra_generation_params,
            "index_of_first_image": self.index_of_first_image,
            "infotexts": self.infotexts,
            "styles": self.styles,
        }

        return json.dumps(obj)