gitea-mirror/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-12-21 23:00:59 -08:00
Code Issues Packages Projects Releases Wiki Activity

add TAESD for i2i and t2i

Browse Source
This commit is contained in:
Kohaku-Blueleaf
2023-08-04 13:38:52 +08:00
parent 3f9e09a615
commit 75336dfc84
5 changed files with 86 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modules/sd_vae_approx.py
View File

@@ -81,6 +81,6 @@ def cheap_approximation(sample):
coefs = torch.tensor(coeffs).to(sample.device)
x_sample = torch.einsum("lxy,lr -> rxy", sample, coefs)
x_sample = torch.einsum("...lxy,lr -> ...rxy", sample, coefs)
return x_sample