Spaces:
Running
on
Zero
Running
on
Zero
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))
|