Cnet (#22)

* ini * remove shit * Create control_model.py * i * i * Update controlnet_supported.py * Update controlnet_supported.py * Update controlnet_supported.py * i * i * Update controlnet_supported.py * i * Update controlnet_supported.py * remove shits * remove shit * Update global_state.py * i * i * Update legacy_preprocessors.py * Update legacy_preprocessors.py * remove shit * Update batch_hijack.py * remove shit * remove shit * i * i * i * Update external_code.py * Update global_state.py * Update infotext.py * Update utils.py * Update external_code.py * i * i * i * Update controlnet_ui_group.py * remove shit * remove shit * i * Update controlnet.py * Update controlnet.py * Update controlnet.py * Update controlnet.py * Update controlnet.py * i * Update global_state.py * Update global_state.py * i * Update global_state.py * Update global_state.py * Update global_state.py * Update global_state.py * Update controlnet_ui_group.py * i * Update global_state.py * Update controlnet_ui_group.py * Update controlnet_ui_group.py * i * Update controlnet_ui_group.py * Update controlnet_ui_group.py * Update controlnet_ui_group.py * Update controlnet_ui_group.py
2024-01-29 14:25:03 -08:00
parent 7ef96eeffa
commit ac374e0b97
35 changed files with 7483 additions and 106 deletions
@@ -0,0 +1,66 @@
+import launch
+import pkg_resources
+import sys
+import os
+import shutil
+import platform
+from pathlib import Path
+from typing import Tuple, Optional
+
+
+repo_root = Path(__file__).parent
+main_req_file = repo_root / "requirements.txt"
+
+
+def comparable_version(version: str) -> Tuple:
+    return tuple(version.split("."))
+
+
+def get_installed_version(package: str) -> Optional[str]:
+    try:
+        return pkg_resources.get_distribution(package).version
+    except Exception:
+        return None
+
+
+def extract_base_package(package_string: str) -> str:
+    base_package = package_string.split("@git")[0]
+    return base_package
+
+
+def install_requirements(req_file):
+    with open(req_file) as file:
+        for package in file:
+            try:
+                package = package.strip()
+                if "==" in package:
+                    package_name, package_version = package.split("==")
+                    installed_version = get_installed_version(package_name)
+                    if installed_version != package_version:
+                        launch.run_pip(
+                            f"install -U {package}",
+                            f"sd-forge-controlnet requirement: changing {package_name} version from {installed_version} to {package_version}",
+                        )
+                elif ">=" in package:
+                    package_name, package_version = package.split(">=")
+                    installed_version = get_installed_version(package_name)
+                    if not installed_version or comparable_version(
+                        installed_version
+                    ) < comparable_version(package_version):
+                        launch.run_pip(
+                            f"install -U {package}",
+                            f"sd-forge-controlnet requirement: changing {package_name} version from {installed_version} to {package_version}",
+                        )
+                elif not launch.is_installed(extract_base_package(package)):
+                    launch.run_pip(
+                        f"install {package}",
+                        f"sd-forge-controlnet requirement: {package}",
+                    )
+            except Exception as e:
+                print(e)
+                print(
+                    f"Warning: Failed to install {package}, some preprocessors may not work."
+                )
+
+
+install_requirements(main_req_file)