technicalkiwi
/
latentblending
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

image->img

This commit is contained in:
Johannes Stelzer 2024-01-07 11:19:09 +01:00
parent d35aacba5f
commit e9ece062a4
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
diffusers_holder.py
View File

@ -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,