From 6d05f40ecb9abeccb925774aaba08a143ade78f4 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Fri, 28 Apr 2023 20:37:59 -0500 Subject: [PATCH] correct param names for VAE encoder wrapper --- api/onnx_web/diffusers/patches/vae.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/onnx_web/diffusers/patches/vae.py b/api/onnx_web/diffusers/patches/vae.py index 975dd803..5cd03483 100644 --- a/api/onnx_web/diffusers/patches/vae.py +++ b/api/onnx_web/diffusers/patches/vae.py @@ -55,7 +55,7 @@ class VAEWrapper(object): if self.decoder: return self.tiled_decode(latent_sample, **kwargs) else: - return self.tiled_encode(latent_sample, **kwargs) + return self.tiled_encode(sample, **kwargs) def __getattr__(self, attr): return getattr(self.wrapped, attr) @@ -106,7 +106,7 @@ class VAEWrapper(object): i : i + self.tile_sample_min_size, j : j + self.tile_sample_min_size, ] - tile = torch.from_numpy(self.wrapped(latent_sample=tile.numpy())[0]) + tile = torch.from_numpy(self.wrapped(sample=tile.numpy())[0]) row.append(tile) rows.append(row) result_rows = []