diff --git a/nodes.py b/nodes.py index 8b75155b6..c558accb2 100644 --- a/nodes.py +++ b/nodes.py @@ -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])