From e9ece062a40d187fddcf5ef12feada3eb36494ee Mon Sep 17 00:00:00 2001 From: Johannes Stelzer Date: Sun, 7 Jan 2024 11:19:09 +0100 Subject: [PATCH] image->img --- diffusers_holder.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/diffusers_holder.py b/diffusers_holder.py index e0d81de..627414a 100644 --- a/diffusers_holder.py +++ b/diffusers_holder.py @@ -46,8 +46,8 @@ class DiffusersHolder(): self.width_latent = self.pipe.unet.config.sample_size self.height_latent = self.pipe.unet.config.sample_size - self.width_image = self.width_latent * self.pipe.vae_scale_factor - self.height_image = self.height_latent * self.pipe.vae_scale_factor + self.width_img = self.width_latent * self.pipe.vae_scale_factor + self.height_img = self.height_latent * self.pipe.vae_scale_factor def init_types(self): assert hasattr(self.pipe, "__class__"), "No valid diffusers pipeline found." @@ -119,8 +119,8 @@ class DiffusersHolder(): latents = self.pipe.prepare_latents( 1, self.pipe.unet.config.in_channels, - self.height_image, - self.width_image, + self.height_img, + self.width_img, torch.float16, self.pipe._execution_device, generator,