66 lines
2.3 KiB
Python
66 lines
2.3 KiB
Python
import os
|
|
import subprocess
|
|
import sys
|
|
import threading
|
|
|
|
root_path = os.path.dirname(__file__)
|
|
parent_dir = os.path.dirname(root_path)
|
|
sys.path.append(root_path)
|
|
from facechain.nodes import *
|
|
from facechain.style_loader_node import *
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"FC StyleLoraLoad": FCStyleLoraLoad,
|
|
"FC FaceDetectCrop": FaceDetectCrop,
|
|
"FC FaceFusion": FCFaceFusion,
|
|
"FC FaceSegment": FCFaceSegment,
|
|
"FC FaceSegAndReplace": FCFaceFusionAndSegReplace,
|
|
"FC RemoveCannyFace": FCRemoveCannyFace,
|
|
"FC CropBottom": FCCropBottom,
|
|
"FC ReplaceByMask": FCReplaceByMask,
|
|
"FC CropAndPaste": FCCropAndPaste,
|
|
"FC MaskOP": FCMaskOP,
|
|
"FC CropToOrigin": FCCropToOrigin,
|
|
}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"FC StyleLoraLoad": "FC StyleLoraLoad",
|
|
"FC FaceDetectCrop": "FC FaceDetectCrop",
|
|
"FC FaceFusion": "FC FaceFusion",
|
|
"FC FaceSegment": "FC FaceSegment",
|
|
"FC FaceSegAndReplace": "FC FaceSegAndReplace",
|
|
"FC RemoveCannyFace": "FC RemoveCannyFace",
|
|
"FC CropBottom": "FC CropBottom",
|
|
"FC ReplaceByMask": "FC ReplaceByMask",
|
|
"FC CropAndPaste": "FC CropAndPaste",
|
|
"FC MaskOP": "FC MaskOP",
|
|
"FC CropToOrigin": "FC CropToOrigin",
|
|
|
|
}
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
|
|
|
|
|
|
# install_model = ["python-slugify==8.0.1", "modelscope", "controlnet_aux==0.0.6", "onnxruntime==1.15.1", "mmcv==1.7.0", "mmdet==2.26.0", "mediapipe==0.10.3", "edge_tts"]
|
|
|
|
def handle_stream(stream, prefix):
|
|
for line in stream:
|
|
print(prefix, line, end="")
|
|
|
|
|
|
def run_script(cmd, cwd='.'):
|
|
process = subprocess.Popen(cmd, cwd=cwd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True, bufsize=1)
|
|
stdout_thread = threading.Thread(target=handle_stream, args=(process.stdout, ""))
|
|
stderr_thread = threading.Thread(target=handle_stream, args=(process.stderr, "[!]"))
|
|
stdout_thread.start()
|
|
stderr_thread.start()
|
|
stdout_thread.join()
|
|
stderr_thread.join()
|
|
return process.wait()
|
|
|
|
|
|
if os.path.basename(parent_dir) == "custom_nodes":
|
|
print("## installing facechain dependencies")
|
|
requirements_path = os.path.join(root_path, "requirements.txt")
|
|
run_script([sys.executable, '-s', '-m', 'pip', 'install', '-q', '-r', requirements_path])
|
|
# for model in install_model:
|
|
# run_script([sys.executable, '-s', '-m', 'pip', 'install', '-q', model])
|