From 79c526f5c092b7e867cc4a860d9329d9bec766cd Mon Sep 17 00:00:00 2001 From: Bradley Reynolds Date: Fri, 10 Jan 2025 16:12:59 -0600 Subject: [PATCH] Only convert to unint8 if directml_enabled --- latent_preview.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/latent_preview.py b/latent_preview.py index b099e4f9..95d3cb73 100644 --- a/latent_preview.py +++ b/latent_preview.py @@ -13,7 +13,8 @@ def preview_to_image(latent_image): latents_ubyte = (((latent_image + 1.0) / 2.0).clamp(0, 1) # change scale from -1..1 to 0..1 .mul(0xFF) # to 0..255 ) - latents_ubyte = latents_ubyte.to(dtype=torch.uint8) + if comfy.model_management.directml_enabled: + latents_ubyte = latents_ubyte.to(dtype=torch.uint8) latents_ubyte = latents_ubyte.to(device="cpu", dtype=torch.uint8, non_blocking=comfy.model_management.device_supports_non_blocking(latent_image.device)) return Image.fromarray(latents_ubyte.numpy())