diff --git a/nodes.py b/nodes.py index 962aa28ec..b817a865e 100644 --- a/nodes.py +++ b/nodes.py @@ -721,8 +721,8 @@ class VAELoader: sd["vae_scale"] = torch.tensor(1.5305) sd["vae_shift"] = torch.tensor(0.0609) elif name == "taef1": - sd["vae_scale"] = torch.tensor(1.5305) - sd["vae_shift"] = torch.tensor(0.0609) + sd["vae_scale"] = torch.tensor(0.3611) + sd["vae_shift"] = torch.tensor(0.1159) return sd @classmethod