-
Notifications
You must be signed in to change notification settings - Fork 21
/
Copy pathinstall.py
29 lines (25 loc) · 1.16 KB
/
install.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
import launch
import os
import pkg_resources
import sys
import traceback
req_file = os.path.join(os.path.dirname(os.path.realpath(__file__)), "requirements.txt")
print("Check FaceSwap requirements")
with open(req_file) as file:
for package in file:
try:
python = sys.executable
package = package.strip()
if not launch.is_installed(package):
print(f"Install {package}")
launch.run_pip(f'install {package}', f"sd-webui-faceswap requirement: {package}")
elif '==' in package:
package_name, package_version = package.split('==')
installed_version = pkg_resources.get_distribution(package_name).version
if installed_version != package_version:
print(f"Install {package}")
launch.run_pip(f'install {package}', f"sd-webui-faceswap requirement: changing {package_name} version from {installed_version} to {package_version}")
except Exception as e:
print(e)
print(f'Warning: Failed to install {package}, faceswap will not work.')
raise e