39 lines
1.7 KiB
Python
39 lines
1.7 KiB
Python
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.')
|