1234567891011121314151617181920212223242526272829303132333435363738 |
- import launch
- import os
- import pkg_resources
- from typing import Tuple, Optional
- req_file = os.path.join(os.path.dirname(os.path.realpath(__file__)), "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
- 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-webui-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-webui-controlnet requirement: changing {package_name} version from {installed_version} to {package_version}")
- elif not launch.is_installed(package):
- launch.run_pip(f"install {package}", f"sd-webui-controlnet requirement: {package}")
- except Exception as e:
- print(e)
- print(f'Warning: Failed to install {package}, some preprocessors may not work.')
|