From 87e39484b801998e78191a488da936992a5a9dde Mon Sep 17 00:00:00 2001 From: patil-suraj Date: Fri, 2 Dec 2022 14:38:25 +0100 Subject: [PATCH] fix uf --- src/diffusers/models/vae.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/models/vae.py b/src/diffusers/models/vae.py index ab8f904e52..7e98281a70 100644 --- a/src/diffusers/models/vae.py +++ b/src/diffusers/models/vae.py @@ -741,7 +741,7 @@ class AutoencoderKL(ModelMixin, ConfigMixin): stride = (min(stride[0], h), min(stride[1], w)) print("reducing stride") - fold, unfold, normalization, weighting = self.get_fold_unfold(z, ks, stride) + fold, unfold, normalization, weighting = self.get_fold_unfold(z, ks, stride, uf=vqf) z = unfold(z) # (bn, nc * prod(**ks), L) # 1. Reshape to img shape