Spaces:
Running
on
A10G
Running
on
A10G
Linoy Tsaban
commited on
Commit
•
4f4effc
1
Parent(s):
343cbb3
Update pipeline_semantic_stable_diffusion_img2img_solver.py
Browse fileschanged 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 |
|