From 4ed942d7f5858d5da2e588540b197f466a5a39d4 Mon Sep 17 00:00:00 2001 From: Robert McCraith Date: Thu, 7 Dec 2023 21:30:04 +0000 Subject: [PATCH] fix: typo in variable name --- stable_diffusion/stable_diffusion/vae.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/stable_diffusion/stable_diffusion/vae.py b/stable_diffusion/stable_diffusion/vae.py index 1b9c8de6..fe473d4c 100644 --- a/stable_diffusion/stable_diffusion/vae.py +++ b/stable_diffusion/stable_diffusion/vae.py @@ -257,7 +257,7 @@ class Autoencoder(nn.Module): def __call__(self, x, key=None): x = self.encoder(x) - x = self.query_proj(x) + x = self.quant_proj(x) mean, logvar = x.split(2, axis=-1) std = mx.exp(0.5 * logvar)