From c72970a917da8dcbf644ebd95cd628fa45462a8e Mon Sep 17 00:00:00 2001 From: Nuked Date: Sun, 19 Nov 2023 08:42:14 +0100 Subject: [PATCH] tests for manager --- install.py | 65 ++++++++++++++---------------------------------- requirements.txt | 1 - 2 files changed, 19 insertions(+), 47 deletions(-) diff --git a/install.py b/install.py index 9f09f22..173f143 100644 --- a/install.py +++ b/install.py @@ -1,57 +1,30 @@ # code based on pysssss repo -import importlib.util import glob import os -import sys from .nnodes import init, get_ext_dir,check_and_install,downloader import folder_paths -import traceback -NODE_CLASS_MAPPINGS = {} -NODE_DISPLAY_NAME_MAPPINGS = {} +py = get_ext_dir("py") +files = glob.glob("*.py", root_dir=py, recursive=False) +check_and_install('py-cpuinfo',"cpuinfo") +check_and_install('gitpython','git') +check_and_install('moviepy') +check_and_install("opencv-python","cv2") +check_and_install('zipfile') +check_and_install('scikit-build',"skbuild") +#LLAMA DEPENTENCIES +check_and_install('typing') +check_and_install('diskcache') +check_and_install('llama_cpp') - - - -if init(): - py = get_ext_dir("py") - files = glob.glob("*.py", root_dir=py, recursive=False) - check_and_install('py-cpuinfo',"cpuinfo") - check_and_install('gitpython','git') - check_and_install('moviepy') - check_and_install("opencv-python","cv2") - check_and_install('zipfile') - check_and_install('scikit-build',"skbuild") - #LLAMA DEPENTENCIES - check_and_install('typing') - check_and_install('diskcache') - check_and_install('llama_cpp') - - - #git clone https://github.com/hzwer/Practical-RIFE.git - from git import Repo - if not os.path.exists(os.path.join(folder_paths.folder_names_and_paths["custom_nodes"][0][0],"ComfyUI-N-Nodes","libs","rifle")): - Repo.clone_from("https://github.com/hzwer/Practical-RIFE.git", os.path.join(folder_paths.folder_names_and_paths["custom_nodes"][0][0],"ComfyUI-N-Nodes","libs","rifle")) - #if train_log folder not exists - if not os.path.exists(os.path.join(folder_paths.folder_names_and_paths["custom_nodes"][0][0],"ComfyUI-N-Nodes","libs","rifle","train_log")): - downloader("https://github.com/Nuked88/DreamingAI/raw/main/RIFE_trained_model_v4.7.zip") +#git clone https://github.com/hzwer/Practical-RIFE.git +from git import Repo +if not os.path.exists(os.path.join(folder_paths.folder_names_and_paths["custom_nodes"][0][0],"ComfyUI-N-Nodes","libs","rifle")): + Repo.clone_from("https://github.com/hzwer/Practical-RIFE.git", os.path.join(folder_paths.folder_names_and_paths["custom_nodes"][0][0],"ComfyUI-N-Nodes","libs","rifle")) +#if train_log folder not exists +if not os.path.exists(os.path.join(folder_paths.folder_names_and_paths["custom_nodes"][0][0],"ComfyUI-N-Nodes","libs","rifle","train_log")): + downloader("https://github.com/Nuked88/DreamingAI/raw/main/RIFE_trained_model_v4.7.zip") - for file in files: - try: - name = os.path.splitext(file)[0] - spec = importlib.util.spec_from_file_location(name, os.path.join(py, file)) - module = importlib.util.module_from_spec(spec) - sys.modules[name] = module - spec.loader.exec_module(module) - if hasattr(module, "NODE_CLASS_MAPPINGS") and getattr(module, "NODE_CLASS_MAPPINGS") is not None: - NODE_CLASS_MAPPINGS.update(module.NODE_CLASS_MAPPINGS) - if hasattr(module, "NODE_DISPLAY_NAME_MAPPINGS") and getattr(module, "NODE_DISPLAY_NAME_MAPPINGS") is not None: - NODE_DISPLAY_NAME_MAPPINGS.update(module.NODE_DISPLAY_NAME_MAPPINGS) - except Exception as e: - traceback.print_exc() - - -__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"] \ No newline at end of file diff --git a/requirements.txt b/requirements.txt index 039bf72..3ae61eb 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,7 +2,6 @@ py-cpuinfo gitpython moviepy opencv-python -zipfile scikit-build typing diskcache \ No newline at end of file