From 07c495feee3889ea2479fb9b4b684f23d00a42ee Mon Sep 17 00:00:00 2001 From: NewLouwa Date: Sun, 19 Oct 2025 03:15:05 +0200 Subject: [PATCH] Fix image handling: Remove dependency on comfy.utils.to_uint8 - Replaced comfy.utils.to_uint8() with standard np.array() for better compatibility - Added missing numpy import in preset_creator_node.py - Functions now work with all ComfyUI versions - Maintained same tensor format [1,H,W,C] for ComfyUI compatibility - Tested and verified working correctly --- nodes/model_preset_loader_node.py | 2 +- nodes/preset_creator_node.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/nodes/model_preset_loader_node.py b/nodes/model_preset_loader_node.py index ef20fde..6938fbb 100644 --- a/nodes/model_preset_loader_node.py +++ b/nodes/model_preset_loader_node.py @@ -236,7 +236,7 @@ def _get_preview_info(model_id: str) -> Dict[str, Any]: def _pil_to_image_tensor(pil: Image.Image) -> torch.Tensor: """Convert PIL Image to ComfyUI tensor format [1,H,W,C]""" img = pil.convert("RGB") - arr = comfy.utils.to_uint8(img) # numpy array uint8 + arr = np.array(img) # numpy array uint8 t = torch.from_numpy(arr).float() / 255.0 return t.unsqueeze(0) # batch = 1 diff --git a/nodes/preset_creator_node.py b/nodes/preset_creator_node.py index 1e7c782..723b45a 100644 --- a/nodes/preset_creator_node.py +++ b/nodes/preset_creator_node.py @@ -10,6 +10,7 @@ from datetime import datetime from typing import Any, Dict, Optional import torch +import numpy as np from PIL import Image import comfy.utils @@ -20,7 +21,7 @@ import folder_paths def _pil_to_image_tensor(pil: Image.Image) -> torch.Tensor: """Convert PIL Image to ComfyUI tensor format [1,H,W,C]""" img = pil.convert("RGB") - arr = comfy.utils.to_uint8(img) # numpy array uint8 + arr = np.array(img) # numpy array uint8 t = torch.from_numpy(arr).float() / 255.0 return t.unsqueeze(0) # batch = 1