diff --git a/install.py b/install.py index 8d0d5ff..ad075cd 100644 --- a/install.py +++ b/install.py @@ -4,14 +4,17 @@ import logging import subprocess import traceback -sys.path.insert(0, os.path.dirname(os.path.abspath(__file__))) +EXT_PATH = os.path.dirname(os.path.abspath(__file__)) + +sys.path.insert(0, EXT_PATH) log = logging.getLogger("AniDoc") download_models = True try: - folder_paths_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..', 'folder_paths.py')) + folder_paths_path = os.path.abspath(os.path.join(os.path.dirname(EXT_PATH), "..", "..", "folder_paths.py")) + sys.path.append(os.path.dirname(folder_paths_path)) import folder_paths @@ -25,8 +28,6 @@ try: except: download_models = False -EXT_PATH = os.path.dirname(os.path.abspath(__file__)) - COTRACKER = os.path.join(EXT_PATH, "cotracker") try: diff --git a/pyproject.toml b/pyproject.toml index 4bb1470..67780a6 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,7 +1,7 @@ [project] name = "comfyui-anidoc" description = "ComfyUI Custom Nodes for 'AniDoc: Animation Creation Made Easier'. This approach automates line art video colorization using a novel model that aligns color information from references, ensures temporal consistency, and reduces manual effort in animation production." -version = "1.0.2" +version = "1.0.3" license = {file = "LICENSE"} dependencies = ["diffusers", "huggingface_hub", "Pillow", "accelerate", "omegaconf", "opencv-python", "einops", "kornia", "git+https://github.com/XPixelGroup/BasicSR.git"]