Only convert to unint8 if directml_enabled

comfyanonymous · Jan 10, 2025 · 79c526f · 79c526f
1 parent 5ed0989
commit 79c526f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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())