44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
"""
|
|
╔════════════════════════════════════════╗
|
|
║ ║
|
|
║ Developed by Abdullah OZMANTAR ║
|
|
║ GitHub : github.com/abdozmantar ║
|
|
║ ║
|
|
╚════════════════════════════════════════╝
|
|
"""
|
|
|
|
import sys
|
|
import os
|
|
|
|
venv_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "venv", "Lib", "site-packages"))
|
|
if os.path.exists(venv_path) and venv_path not in sys.path:
|
|
sys.path.insert(0, venv_path)
|
|
|
|
repo_dir = os.path.dirname(os.path.realpath(__file__))
|
|
sys.path.insert(0, repo_dir)
|
|
original_modules = sys.modules.copy()
|
|
|
|
|
|
modules_used = [
|
|
"demucs",
|
|
"demucs.demucs",
|
|
"demucs.hdemucs",
|
|
"demucs.htdemucs",
|
|
"utils",
|
|
]
|
|
original_webui_modules = {}
|
|
for module in modules_used:
|
|
if module in sys.modules:
|
|
original_webui_modules[module] = sys.modules.pop(module)
|
|
|
|
from .node import DeepExtractV2Node
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"DeepExtractV2Node": DeepExtractV2Node
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"DeepExtractV2Node": "🎵 Deep Extract V2 - Separator"
|
|
}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"] |