diff --git a/__init__.py b/__init__.py index c0bb79e..dc076fe 100644 --- a/__init__.py +++ b/__init__.py @@ -39,6 +39,7 @@ try: from mmdet.evaluation import get_classes from skimage.measure import label, regionprops from collections import namedtuple + import piexif except: import importlib print("### ComfyUI-Impact-Pack: Reinstall dependencies (several dependencies are missing.)") diff --git a/modules/impact/impact_pack.py b/modules/impact/impact_pack.py index b465542..bf2a508 100644 --- a/modules/impact/impact_pack.py +++ b/modules/impact/impact_pack.py @@ -5,6 +5,10 @@ import comfy.samplers import comfy.sd import warnings from segment_anything import sam_model_registry +from io import BytesIO +import piexif +import zipfile +from server import PromptServer from impact.utils import * import impact.core as core @@ -1497,11 +1501,6 @@ class ImageSender(nodes.PreviewImage): return result -from io import BytesIO -import piexif -import zipfile -from server import PromptServer - class LatentReceiver: def __init__(self): self.input_dir = folder_paths.get_input_directory() diff --git a/requirements.txt b/requirements.txt index 3ebff67..6dfa813 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,4 @@ openmim segment-anything scikit-image +piexif \ No newline at end of file