This commit is contained in:
Lt.Dr.Data 2023-06-08 18:21:25 +09:00
parent 86755c9f80
commit 0a0afaf3c8

View File

@ -329,12 +329,9 @@ class SavePreviewLatent(SaveLatent):
@staticmethod
def prepare_preview(latent_tensor):
lower_bound = 128
upper_bound = 512
device = comfy.model_management.get_torch_device()
previewer = latent_preview.get_previewer(device, True)
upper_bound = 256
previewer = latent_preview.get_previewer("cpu", True)
image = previewer.decode_latent_to_preview(latent_tensor)
min_size = min(image.size[0], image.size[1])
max_size = max(image.size[0], image.size[1])