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.width_latent = self.pipe.unet.config.sample_size
self.height_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.width_img = self.width_latent * self.pipe.vae_scale_factor
self.height_image = self.height_latent * self.pipe.vae_scale_factor self.height_img = self.height_latent * self.pipe.vae_scale_factor
def init_types(self): def init_types(self):
assert hasattr(self.pipe, "__class__"), "No valid diffusers pipeline found." assert hasattr(self.pipe, "__class__"), "No valid diffusers pipeline found."
@ -119,8 +119,8 @@ class DiffusersHolder():
latents = self.pipe.prepare_latents( latents = self.pipe.prepare_latents(
1, 1,
self.pipe.unet.config.in_channels, self.pipe.unet.config.in_channels,
self.height_image, self.height_img,
self.width_image, self.width_img,
torch.float16, torch.float16,
self.pipe._execution_device, self.pipe._execution_device,
generator, generator,