From 604d34900a165edb4b75f3900842079bd345fe53 Mon Sep 17 00:00:00 2001
From: wl-zhao <zhaowl20@mails.tsinghua.edu.cn>
Date: Mon, 16 Dec 2024 17:25:23 +0800
Subject: [PATCH] support heif image

---
 comfy-nodes/input_image.py | 3 +++
 requirements.txt           | 3 ++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/comfy-nodes/input_image.py b/comfy-nodes/input_image.py
index 9896832..9df4a00 100755
--- a/comfy-nodes/input_image.py
+++ b/comfy-nodes/input_image.py
@@ -7,6 +7,9 @@ import torch
 import os
 import uuid
 import tqdm
+from pillow_heif import register_heif_opener
+
+register_heif_opener()
 
 
 class ShellAgentPluginInputImage:
diff --git a/requirements.txt b/requirements.txt
index f1d3a61..6666fe6 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,4 +3,5 @@ pydantic
 opencv-python
 imageio-ffmpeg
 brotli
-# logfire
\ No newline at end of file
+pillow_heif
+# logfire