Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -753,7 +753,7 @@ def generate_image(
|
|
| 753 |
x = t * x + (1.0 - t) * init_image.to(x.dtype)
|
| 754 |
|
| 755 |
inp = prepare(t5=t5, clip=clip, img=x, prompt=[neg_prompt, prompt])
|
| 756 |
-
x = denoise(model, **inp, timesteps=timesteps, guidance=guidance, use_cfg_guidance=True,
|
| 757 |
|
| 758 |
# with profile(activities=[ProfilerActivity.CPU],record_shapes=True,profile_memory=True) as prof:
|
| 759 |
# print(prof.key_averages().table(sort_by="cpu_time_total", row_limit=20))
|
|
|
|
| 753 |
x = t * x + (1.0 - t) * init_image.to(x.dtype)
|
| 754 |
|
| 755 |
inp = prepare(t5=t5, clip=clip, img=x, prompt=[neg_prompt, prompt])
|
| 756 |
+
x = denoise(model, **inp, timesteps=timesteps, guidance=guidance, use_cfg_guidance=True, joint_attention_kwargs=self.joint_attention_kwargs)
|
| 757 |
|
| 758 |
# with profile(activities=[ProfilerActivity.CPU],record_shapes=True,profile_memory=True) as prof:
|
| 759 |
# print(prof.key_averages().table(sort_by="cpu_time_total", row_limit=20))
|