Spaces:
Running
on
A10G
Running
on
A10G
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
|