42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
import os
|
|
import sys
|
|
import subprocess
|
|
|
|
comfy_path = '../..'
|
|
if sys.argv[0] == 'install.py':
|
|
sys.path.append('.') # for portable version
|
|
|
|
impact_path = os.path.join(os.path.dirname(__file__), "modules")
|
|
|
|
sys.path.append(impact_path)
|
|
sys.path.append(comfy_path)
|
|
|
|
import platform
|
|
import folder_paths
|
|
from torchvision.datasets.utils import download_url
|
|
|
|
print("### ComfyUI-FaceSwap: Check dependencies")
|
|
|
|
if "python_embeded" in sys.executable or "python_embedded" in sys.executable:
|
|
pip_install = [sys.executable, '-s', '-m', 'pip', 'install']
|
|
else:
|
|
pip_install = [sys.executable, '-m', 'pip', 'install']
|
|
|
|
def ensure_pip_packages():
|
|
try:
|
|
import cython
|
|
except Exception:
|
|
my_path = os.path.dirname(__file__)
|
|
requirements_path = os.path.join(my_path, "requirements.txt")
|
|
subprocess.check_call(pip_install + ['-r', requirements_path])
|
|
|
|
def install():
|
|
ensure_pip_packages()
|
|
# Download model
|
|
print("### ComfyUI-FaceSwap: Check basic models")
|
|
model_path = folder_paths.models_dir
|
|
onnx_path = os.path.join(model_path, "roop")
|
|
|
|
if not os.path.exists(onnx_path):
|
|
download_url("https://huggingface.co/henryruhs/roop/resolve/main/inswapper_128.onnx", onnx_path)
|