diff --git a/install.py b/install.py index d68693c..8d0d5ff 100644 --- a/install.py +++ b/install.py @@ -11,7 +11,11 @@ log = logging.getLogger("AniDoc") download_models = True try: + folder_paths_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..', 'folder_paths.py')) + sys.path.append(os.path.dirname(folder_paths_path)) + import folder_paths + DIFFUSERS_DIR = os.path.join(folder_paths.models_dir, "diffusers") ANIDOC_DIR = os.path.join(DIFFUSERS_DIR, "anidoc") SVD_I2V_DIR = os.path.join( @@ -20,7 +24,6 @@ try: ) except: download_models = False - log.info("Not called by ComfyUI Manager. Models will not be downloaded") EXT_PATH = os.path.dirname(os.path.abspath(__file__)) diff --git a/pyproject.toml b/pyproject.toml index 012533d..4bb1470 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.1" +version = "1.0.2" license = {file = "LICENSE"} dependencies = ["diffusers", "huggingface_hub", "Pillow", "accelerate", "omegaconf", "opencv-python", "einops", "kornia", "git+https://github.com/XPixelGroup/BasicSR.git"]