From e6bd5af6a8e306a1cdef63402a77a980a04ad6e1 Mon Sep 17 00:00:00 2001
From: Grafting Rayman <156515434+GraftingRayman@users.noreply.github.com>
Date: Fri, 17 Jan 2025 11:06:44 +0000
Subject: Add files via upload

---
 install.py | 104 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 104 insertions(+)
 create mode 100644 install.py

(limited to 'install.py')

diff --git a/install.py b/install.py
new file mode 100644
index 0000000..bf4f6f2
--- /dev/null
+++ b/install.py
@@ -0,0 +1,104 @@
+import warnings
+warnings.filterwarnings("ignore", category=DeprecationWarning)
+
+import subprocess
+import os, sys
+try:
+    from pkg_resources import get_distribution as distributions
+except:
+    from importlib_metadata import distributions
+from tqdm import tqdm
+import urllib.request
+from packaging import version as pv
+try:
+    from folder_paths import models_dir
+except:
+    from pathlib import Path
+    models_dir = os.path.join(Path(__file__).parents[2], "models")
+
+sys.path.append(os.path.dirname(os.path.realpath(__file__)))
+
+req_file = os.path.join(os.path.dirname(os.path.realpath(__file__)), "requirements.txt")
+
+model_url = "https://huggingface.co/datasets/Gourieff/ReActor/resolve/main/models/inswapper_128.onnx"
+model_name = os.path.basename(model_url)
+models_dir_path = os.path.join(models_dir, "insightface")
+model_path = os.path.join(models_dir_path, model_name)
+
+def run_pip(*args):
+    subprocess.run([sys.executable, "-m", "pip", "install", "--no-warn-script-location", *args])
+
+def is_installed (
+        package: str, version: str = None, strict: bool = True
+):
+    has_package = None
+    try:
+        has_package = distributions(package)
+        if has_package is not None:
+            if version is not None:
+                installed_version = has_package.version
+                if (installed_version != version and strict == True) or (pv.parse(installed_version) < pv.parse(version) and strict == False):
+                    return False
+                else:
+                    return True
+            else:
+                return True
+        else:
+            return False
+    except Exception as e:
+        print(f"Status: {e}")
+        return False
+    
+def download(url, path, name):
+    request = urllib.request.urlopen(url)
+    total = int(request.headers.get('Content-Length', 0))
+    with tqdm(total=total, desc=f'[ReActor] Downloading {name} to {path}', unit='B', unit_scale=True, unit_divisor=1024) as progress:
+        urllib.request.urlretrieve(url, path, reporthook=lambda count, block_size, total_size: progress.update(block_size))
+
+if not os.path.exists(models_dir_path):
+    os.makedirs(models_dir_path)
+
+if not os.path.exists(model_path):
+    download(model_url, model_path, model_name)
+
+with open(req_file) as file:
+    try:
+        ort = "onnxruntime-gpu"
+        import torch
+        cuda_version = None
+        if torch.cuda.is_available():
+            cuda_version = torch.version.cuda
+            print(f"CUDA {cuda_version}")
+        elif torch.backends.mps.is_available() or hasattr(torch,'dml') or hasattr(torch,'privateuseone'):
+            ort = "onnxruntime"
+        if cuda_version is not None and float(cuda_version)>=12 and torch.torch_version.__version__ <= "2.2.0": # CU12.x and torch<=2.2.0
+            print(f"Torch: {torch.torch_version.__version__}")
+            if not is_installed(ort,"1.17.0",False):
+                run_pip(ort,"--extra-index-url", "https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/")
+        elif cuda_version is not None and float(cuda_version)>=12 and torch.torch_version.__version__ >= "2.4.0" : # CU12.x and latest torch
+            print(f"Torch: {torch.torch_version.__version__}")
+            if not is_installed(ort,"1.20.1",False): # latest ort-gpu
+                run_pip(ort,"-U")
+        elif not is_installed(ort,"1.16.1",False):
+            run_pip(ort, "-U")
+    except Exception as e:
+        print(e)
+        print(f"Warning: Failed to install {ort}, ReActor will not work.")
+        raise e
+    strict = True
+    for package in file:
+        package_version = None
+        try:
+            package = package.strip()
+            if "==" in package:
+                package_version = package.split('==')[1]
+            elif ">=" in package:
+                package_version = package.split('>=')[1]
+                strict = False
+            if not is_installed(package,package_version,strict):
+                run_pip(package)
+        except Exception as e:
+            print(e)
+            print(f"Warning: Failed to install {package}, ReActor will not work.")
+            raise e
+print("Ok")
-- 
cgit v1.2.3