Merge 79c526f5c0 into 2ff3104f70

Only convert to unint8 if directml_enabled
Convert latents_ubyte to 8-bit unsigned int before converting to CPU
2025-01-11 02:15:17 +00:00 · 2025-01-10 22:13:02 +00:00 · 2025-01-10 16:12:59 -06:00 · 2024-12-31 15:13:23 -06:00
1 changed files with 4 additions and 1 deletions
--- a/latent_preview.py
+++ b/latent_preview.py
@ -12,7 +12,10 @@ MAX_PREVIEW_RESOLUTION = args.preview_size
 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
-                            ).to(device="cpu", dtype=torch.uint8, non_blocking=comfy.model_management.device_supports_non_blocking(latent_image.device))
+                            )
+        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())
Author	SHA1	Message	Date
Bradley Reynolds	044939b596	Merge `79c526f5c0` into `2ff3104f70`	2025-01-10 22:13:02 +00:00
Bradley Reynolds	79c526f5c0	Only convert to unint8 if directml_enabled	2025-01-10 16:12:59 -06:00
Bradley Reynolds	5ed09898bb	Convert latents_ubyte to 8-bit unsigned int before converting to CPU	2024-12-31 15:13:23 -06:00