Spaces:
Running
on
A10G
Running
on
A10G
Linoy Tsaban
commited on
Commit
•
8f22003
1
Parent(s):
68f9d7d
Update pipeline_semantic_stable_diffusion_img2img_solver.py
Browse files
pipeline_semantic_stable_diffusion_img2img_solver.py
CHANGED
|
@@ -1058,7 +1058,7 @@ class SemanticStableDiffusionImg2ImgPipeline_DPMSolver(DiffusionPipeline):
|
|
| 1058 |
self.scheduler.num_inference_steps = timesteps.shape[0]
|
| 1059 |
self.scheduler.timesteps = timesteps
|
| 1060 |
|
| 1061 |
-
|
| 1062 |
# 1. get embeddings
|
| 1063 |
|
| 1064 |
uncond_embedding = self.encode_text("")
|
|
|
|
| 1058 |
self.scheduler.num_inference_steps = timesteps.shape[0]
|
| 1059 |
self.scheduler.timesteps = timesteps
|
| 1060 |
|
| 1061 |
+
self.unet.set_attn_processor(AttnProcessor())
|
| 1062 |
# 1. get embeddings
|
| 1063 |
|
| 1064 |
uncond_embedding = self.encode_text("")
|