Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

lint cleanup

Browse Source
This commit is contained in:
Vladimir Mandic
2024-05-28 10:48:27 -04:00
parent d7c732bfe6
commit 9a7a5ba81c
20 changed files with 68 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_class.py
+2
View File
@@ -419,6 +419,8 @@ class StableDiffusionProcessingImg2Img(StableDiffusionProcessing):
batch_images = np.expand_dims(self.init_images[0], axis=0).repeat(self.batch_size, axis=0)
elif len(self.init_images) <= self.batch_size:
batch_images = np.array(self.init_images)
else:
batch_images = np.array(self.init_images[:self.batch_size])
image = torch.from_numpy(batch_images)
image = 2. * image - 1.
image = image.to(device=shared.device, dtype=devices.dtype_vae)