hyoungwoncho
/

sd_perturbed_attention_guidance

Unconditional Image Generation

Diffusion Models

Stable Diffusion

Perturbed-Attention Guidance

Model card Files Files and versions Community

hyoungwoncho commited on Apr 16

Commit

5193ae7

•

1 Parent(s): 30c94b4

Update README.md

Files changed (1) hide show

README.md +4 -1

README.md CHANGED Viewed

@@ -27,11 +27,14 @@ from diffusers import StableDiffusionPipeline
 pipe = StableDiffusionPipeline.from_pretrained(
     "runwayml/stable-diffusion-v1-5",
     custom_pipeline="hyoungwoncho/sd_perturbed_attention_guidance",
-    torch_dtype=torch.float16
 )
 device="cuda"
 pipe = pipe.to(device)
 ```
 Sampling with PAG:

 pipe = StableDiffusionPipeline.from_pretrained(
     "runwayml/stable-diffusion-v1-5",
     custom_pipeline="hyoungwoncho/sd_perturbed_attention_guidance",
+    torch_dtype=torch.float16,
+    safety_checker=None
 )
 device="cuda"
 pipe = pipe.to(device)
+prompts = ["a corgi"]
 ```
 Sampling with PAG: