fix numbers

This commit is contained in:
Matthew Turnshek 2024-08-16 12:43:12 -04:00
parent 17b91ae596
commit 004b68e707

View File

@ -721,8 +721,8 @@ class VAELoader:
sd["vae_scale"] = torch.tensor(1.5305) sd["vae_scale"] = torch.tensor(1.5305)
sd["vae_shift"] = torch.tensor(0.0609) sd["vae_shift"] = torch.tensor(0.0609)
elif name == "taef1": elif name == "taef1":
sd["vae_scale"] = torch.tensor(1.5305) sd["vae_scale"] = torch.tensor(0.3611)
sd["vae_shift"] = torch.tensor(0.0609) sd["vae_shift"] = torch.tensor(0.1159)
return sd return sd
@classmethod @classmethod