Is it possible to add variables to the X/Y script?

[tusharbhutt]

As per subject. Is it possible to add variables that are not in the script already? Ithink this is where they'd go:

axis_options = [
    AxisOption("Nothing", str, do_nothing, format_nothing, None),
    AxisOption("Seed", int, apply_field("seed"), format_value_add_label, None),
    AxisOption("Var. seed", int, apply_field("subseed"), format_value_add_label, None),
    AxisOption("Var. strength", float, apply_field("subseed_strength"), format_value_add_label, None),

etc.

But, I don't know the names of the variables. Is there a list of variables to look at? I'd like to add HEIGHT and WIDTH as well as embeddings.

Thanks,

TB

[xram64]

The first string in each AxisOption (the label) is just for display in the dropdown and doesn't correspond to anything. What matters most is the third argument, which indicates a function to handle making changes to the generation variables before processing. For instance, in the "Seed" option, apply_field("seed") will become a function that changes the seed property of the next image to be processed.

You can see a few functions like this at the top of the script, which each apply to a different kind of property. Some of them like apply_hypernetwork or apply_sampler are specific to one parameter. But for the ones that use apply_field, the place to look for the variable names is in the StableDiffusionProcessing class in modules/processing.py (this is the class that the p variable belongs to, which holds the generation params). There, you can see names like self.seed, self.height, self.width, etc.

So for some of these variables (height/width in particular), all it should take is adding the option to the list, and making sure to provide the correct type in the second argument:

AxisOption("Height", int, apply_field("height"), format_value_add_label, None),
AxisOption("Width", int, apply_field("width"), format_value_add_label, None),

[tusharbhutt]

Thank you! I'll give that a shot and then look around for the embedding variable name as well

[tusharbhutt]

Sorry to bother you @xram64 but would you know how to grab a variable from another script? I've tried this for aesthetic variable,s but it doesn't seem to grab or adjust them, it just is happy to plot the same value (if it even gets them) over and over again.

This is what I put in the X/Y script:

    AxisOption("Aes_Learn", float, apply_field("aesthetic_lr"), format_value_add_label, None),
    AxisOption("Aes_Weight", float, apply_field("aesthetic_weight"), format_value_add_label, None),	
    AxisOption("Aes_Steps", int, apply_field("aesthetic_steps"), format_value_add_label, None),	

I'm guessing it has to call the aesthetic script on each loop but not sure how to do that.

[MartinCairnsSQL]

I've done a PR to add the aesthetic gradient parameters #4000

[xram64]

Nice, the PR should make things easier.

I couldn't find a simple way to do this from just modifying the X/Y script, but I did make it work with a small change to the extension itself.
So as a workaround, you can just replace the process() function (lines 35-36) in extensions/aesthetic-gradients/scripts/aesthetic.py with this:

    def process(self, p, aesthetic_weight, aesthetic_steps, aesthetic_lr, aesthetic_slerp, aesthetic_imgs, aesthetic_imgs_text, aesthetic_slerp_angle, aesthetic_text_negative):
        lr, weight, steps = float(aesthetic_lr), float(aesthetic_weight), int(aesthetic_steps)
        
        if hasattr(p, 'aesthetic_lr'):
            lr = float(p.aesthetic_lr)
        if hasattr(p, 'aesthetic_weight'):
            weight = float(p.aesthetic_weight)
        if hasattr(p, 'aesthetic_steps'):
            steps = int(p.aesthetic_steps)
        
        aesthetic.set_aesthetic_params(p, lr, weight, steps, aesthetic_imgs, aesthetic_slerp, aesthetic_imgs_text, aesthetic_slerp_angle, aesthetic_text_negative)

[xram64]

(That is, once you've already added the same 3 lines to xy_grid.py that you posted:)

    AxisOption("Aes_Learn", float, apply_field("aesthetic_lr"), format_value_add_label, None),
    AxisOption("Aes_Weight", float, apply_field("aesthetic_weight"), format_value_add_label, None),
    AxisOption("Aes_Steps", int, apply_field("aesthetic_steps"), format_value_add_label, None),