69 lines
2.4 KiB
Python
69 lines
2.4 KiB
Python
import os
|
|
from time import sleep
|
|
from importlib.util import spec_from_file_location, module_from_spec
|
|
import sys
|
|
import argparse
|
|
import subprocess
|
|
|
|
this_module_name = "ComfyUI_Lam"
|
|
EXT_PATH = os.path.dirname(os.path.realpath(__file__))
|
|
sys.path.append(os.path.abspath(os.path.join(EXT_PATH, "../../")))
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--no_download_ckpts', action="store_true", help="Don't download any model")
|
|
|
|
args = parser.parse_args()
|
|
|
|
def add_global_shortcut_module(module_name, module_path):
|
|
#Naming things is hard
|
|
module_spec = spec_from_file_location(module_name, module_path)
|
|
module = module_from_spec(module_spec)
|
|
sys.modules[module_name] = module
|
|
module_spec.loader.exec_module(module)
|
|
|
|
def download_models():
|
|
#from git.repo import Repo
|
|
#import folder_paths
|
|
#download_path = os.path.join(folder_paths.models_dir,'image-face-fusion')
|
|
#Repo.clone_from('https://www.modelscope.cn/damo/cv_unet_face_fusion_torch.git',to_path=download_path,branch='master')
|
|
pass
|
|
|
|
|
|
# command1 = [sys.executable]
|
|
# if "python_embeded" in sys.executable: command1 += ['-s']
|
|
# command1 += ['-m','pip', 'install','-U', 'pip']
|
|
# print("upgrade pip...")
|
|
# sleep(2)
|
|
# proc = subprocess.Popen(command1, stdout=subprocess.PIPE, bufsize=1, universal_newlines=True)
|
|
# for line in iter(proc.stdout.readline, ''):
|
|
# print(line, end='')
|
|
# proc.wait()
|
|
# print("upgrade pip done")
|
|
|
|
command = [sys.executable]
|
|
if "python_embeded" in sys.executable: command += ['-s']
|
|
command += ['-m','pip', 'install','--use-pep517', '-r' , f'{EXT_PATH}/requirements.txt',
|
|
'--index-url' , 'https://mirrors.aliyun.com/pypi/simple',
|
|
'--extra-index-url', 'https://download.pytorch.org/whl/cu117', '--no-warn-script-location']
|
|
print("Installing requirements...")
|
|
sleep(2)
|
|
proc = subprocess.Popen(command, stdout=subprocess.PIPE, bufsize=1, universal_newlines=True)
|
|
for line in iter(proc.stdout.readline, ''):
|
|
print(line, end='')
|
|
proc.wait()
|
|
print("Installing requirements done")
|
|
|
|
if args.no_download_ckpts: exit()
|
|
print("Download models...")
|
|
download_models()
|
|
print("Download done")
|
|
|
|
# add_global_shortcut_module("cli_args", os.path.join(EXT_PATH, "../../comfy/cli_args.py"))
|
|
# add_global_shortcut_module("model_management", os.path.join(EXT_PATH, "../../comfy/model_management.py"))
|
|
# add_global_shortcut_module(this_module_name, os.path.join(EXT_PATH, "__init__.py"))
|
|
|
|
|
|
sleep(2)
|
|
download_models()
|
|
print("Done!")
|