diff --git a/py/imagefunc.py b/py/imagefunc.py index 9790341..4a4c7a7 100644 --- a/py/imagefunc.py +++ b/py/imagefunc.py @@ -137,7 +137,15 @@ def tensor2np(tensor: torch.Tensor) -> List[np.ndarray]: return [np.clip(255.0 * t.cpu().numpy(), 0, 255).astype(np.uint8) for t in tensor] def tensor2pil(t_image: torch.Tensor) -> Image: - return Image.fromarray(np.clip(255.0 * t_image.cpu().numpy().squeeze(), 0, 255).astype(np.uint8)) + if t_image.dtype != torch.float32: + t_image = t_image.float() + return Image.fromarray( + np.clip( + 255.0 * t_image.cpu().numpy().squeeze(), + 0, + 255 + ).astype(np.uint8) + ) def tensor2cv2(image:torch.Tensor) -> np.array: if image.dim() == 4: diff --git a/pyproject.toml b/pyproject.toml index 4ad7dcf..226bf06 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,7 +1,7 @@ [project] name = "ComfyUI_LayerStyle_Advance" description = "The nodes detached from ComfyUI Layer Style are mainly those with complex requirements for dependency packages." -version = "2.0.32" +version = "2.0.33" license = { text = "MIT License" } dependencies = ["numpy", "matplotlib", "scikit_image", "scikit_learn", "opencv-contrib-python", "pymatting", "timm", "blend_modes", "transformers", "diffusers", "loguru", "colour-science", "huggingface_hub", "segment_anything", "addict", "omegaconf", "yapf", "wget", "iopath", "mediapipe", "typer_config", "fastapi", "rich", "google-generativeai", "ultralytics", "transparent-background", "accelerate", "onnxruntime", "bitsandbytes", "peft", "protobuf", "hydra-core", "blind-watermark", "qrcode", "pyzbar", "psd-tools", "wandb", "zhipuai", "openai","google-genai", "fastapi","typer-config"]