google / style-aligned

Official code for "Style Aligned Image Generation via Shared Attention"
Apache License 2.0
1.16k stars 83 forks source link

run in free colab #5

Open loboere opened 7 months ago

loboere commented 7 months ago

Is there a way to make it work in free colab?

a-r-r-o-w commented 7 months ago

You can checkout a quick hacky notebook I made here if you'd like to run on colab.

Colab

a-r-r-o-w commented 7 months ago

Also this colab: https://github.com/google/style-aligned/issues/3#issuecomment-1843197744

demiguelm commented 7 months ago

You can checkout a quick hacky notebook I made here if you'd like to run on colab.

Colab

Great!You aded some parameters like steps, negative prompt and so on, any possibility to choose the sampler?

a-r-r-o-w commented 7 months ago

You could try doing it like so:

pipe = ... # Initialize the pipeline

from diffusers.schedulers import DPMSolverMultistepScheduler
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)

This notebook is a bit restricted and doesn't support everything the authors provide here (for example, controlnets). I'm working on improving it so that their work can be used with any pipeline.

demiguelm commented 7 months ago

You could try doing it like so:

pipe = ... # Initialize the pipeline

from diffusers.schedulers import DPMSolverMultistepScheduler
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)

This notebook is a bit restricted and doesn't support everything the authors provide here (for example, controlnets). I'm working on improving it so that their work can be used with any pipeline.

Confy or A1111 maybe?