remove TODO

This commit is contained in:
Matthew Turnshek 2024-08-16 12:06:56 -04:00
parent 1c14fc89db
commit 17b91ae596

View File

@ -721,7 +721,6 @@ 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":
# TODO: check these numbers
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)
return sd return sd