Spaces:
Runtime error
Runtime error
Commit
·
ad8e547
1
Parent(s):
daadb28
Update pipeline_semantic_stable_diffusion_img2img_solver.py
Browse files
pipeline_semantic_stable_diffusion_img2img_solver.py
CHANGED
|
@@ -601,6 +601,8 @@ class SemanticStableDiffusionImg2ImgPipeline_DPMSolver(DiffusionPipeline):
|
|
| 601 |
second element is a list of `bool`s denoting whether the corresponding generated image likely represents
|
| 602 |
"not-safe-for-work" (nsfw) content, according to the `safety_checker`.
|
| 603 |
"""
|
|
|
|
|
|
|
| 604 |
# eta = self.eta
|
| 605 |
num_images_per_prompt = 1
|
| 606 |
# latents = self.init_latents
|
|
|
|
| 601 |
second element is a list of `bool`s denoting whether the corresponding generated image likely represents
|
| 602 |
"not-safe-for-work" (nsfw) content, according to the `safety_checker`.
|
| 603 |
"""
|
| 604 |
+
self.unet.set_attn_processor(AttnProcessor())
|
| 605 |
+
|
| 606 |
# eta = self.eta
|
| 607 |
num_images_per_prompt = 1
|
| 608 |
# latents = self.init_latents
|