File size: 1,165 Bytes
b27d22d b8f0ab3 b27d22d bc0a907 9085f30 bc0a907 b27d22d c9b62a3 a81fa9d b27d22d bc0a907 b27d22d 58d4985 bc0a907 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
---
language:
- en
pipeline_tag: unconditional-image-generation
tags:
- Diffusion Models
- Stable Diffusion
- Perturbed-Attention Guidance
- PAG
---
# Perturbed-Attention Guidance
This repository is based on [Diffusers](https://huggingface.co/docs/diffusers/index). StableDiffusionPAGPipeline is a modification of StableDiffusionPipeline to support Perturbed-Attention Guidance (PAG).
## Quickstart
Load StableDiffusionPAGPipeline as below:
```
pipe = StableDiffusionPipeline.from_pretrained(
"runwayml/stable-diffusion-v1-5",
custom_pipeline="hyoungwoncho/sd_perturbed_attention_guidance",
torch_dtype=torch.float16,
safety_checker=None
)
```
Sampling:
```
output_baseline = pipe(
prompts,
width=512,
height=512,
num_inference_steps=50,
guidance_scale=0.0,
pag_scale=5.0,
pag_applied_layers_index=['m0']
).images
```
## Parameters
pag_scale : gudiance scale of PAG (ex: 4.5)
pag_applied_layers_index = index of the layer to apply perturbation (ex: ['m0'])
|