Linoy Tsaban commited on
Commit
4f4effc
1 Parent(s): 343cbb3

Update pipeline_semantic_stable_diffusion_img2img_solver.py

Browse files

changed attention resetting to be compatible with diffusers version

pipeline_semantic_stable_diffusion_img2img_solver.py CHANGED
@@ -1160,7 +1160,8 @@ class SemanticStableDiffusionImg2ImgPipeline_DPMSolver(DiffusionPipeline):
1160
  self.scheduler.timesteps = timesteps
1161
 
1162
  # Reset attn processor, we do not want to store attn maps during inversion
1163
- self.unet.set_default_attn_processor()
 
1164
 
1165
  # 1. get embeddings
1166
 
 
1160
  self.scheduler.timesteps = timesteps
1161
 
1162
  # Reset attn processor, we do not want to store attn maps during inversion
1163
+ # self.unet.set_default_attn_processor()
1164
+ self.unet.set_attn_processor(AttnProcessor())
1165
 
1166
  # 1. get embeddings
1167