multimodalart HF staff commited on
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