diff --git a/install.py b/install.py index 94ccdd0..acb6182 100644 --- a/install.py +++ b/install.py @@ -1,15 +1,20 @@ import launch - import os +import pkg_resources req_file = os.path.join(os.path.dirname(os.path.realpath(__file__)), "requirements.txt") with open(req_file) as file: - - for lib in file: - - lib = lib.strip() - - if not launch.is_installed(lib): - - launch.run_pip(f"install {lib}", f"SD-CN-Animation requirement: {lib}") + for package in file: + try: + package = package.strip() + if '==' in package: + package_name, package_version = package.split('==') + installed_version = pkg_resources.get_distribution(package_name).version + if installed_version != package_version: + launch.run_pip(f"install {package}", f"SD-CN-Animation 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-CN-Animation requirement: {package}") + except Exception as e: + print(e) + print(f'Warning: Failed to install {package}.') \ No newline at end of file diff --git a/readme.md b/readme.md index 3af10cf..0ebbd50 100644 --- a/readme.md +++ b/readme.md @@ -82,4 +82,5 @@ pip install sckit-image==0.19.2 --no-cache-dir