TAESD fix

This commit is contained in:
Sakura-Luna
2023-05-17 17:39:07 +08:00
parent 85232a5b26
commit 7a13a3f4ba
2 changed files with 6 additions and 5 deletions

View File

@@ -45,7 +45,7 @@ def decoder():
class TAESD(nn.Module):
latent_magnitude = 2
latent_magnitude = 3
latent_shift = 0.5
def __init__(self, decoder_path="taesd_decoder.pth"):