diff options
| author | Grafting Rayman <156515434+GraftingRayman@users.noreply.github.com> | 2025-01-17 11:06:44 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-01-17 11:06:44 +0000 |
| commit | e6bd5af6a8e306a1cdef63402a77a980a04ad6e1 (patch) | |
| tree | d0732226bbc22feedad9e834b2218d7d0b0eff54 /__init__.py | |
| parent | 495ffc4777522e40941753e3b1b79c02f84b25b4 (diff) | |
| download | Comfyui-reactor-node-e6bd5af6a8e306a1cdef63402a77a980a04ad6e1.tar.gz | |
Diffstat (limited to '__init__.py')
| -rw-r--r-- | __init__.py | 39 |
1 files changed, 39 insertions, 0 deletions
diff --git a/__init__.py b/__init__.py new file mode 100644 index 0000000..3f8982d --- /dev/null +++ b/__init__.py @@ -0,0 +1,39 @@ +import sys
+import os
+
+repo_dir = os.path.dirname(os.path.realpath(__file__))
+sys.path.insert(0, repo_dir)
+original_modules = sys.modules.copy()
+
+# Place aside existing modules if using a1111 web ui
+modules_used = [
+ "modules",
+ "modules.images",
+ "modules.processing",
+ "modules.scripts_postprocessing",
+ "modules.scripts",
+ "modules.shared",
+]
+original_webui_modules = {}
+for module in modules_used:
+ if module in sys.modules:
+ original_webui_modules[module] = sys.modules.pop(module)
+
+# Proceed with node setup
+from .nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
+
+__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
+
+# Clean up imports
+# Remove repo directory from path
+sys.path.remove(repo_dir)
+# Remove any new modules
+modules_to_remove = []
+for module in sys.modules:
+ if module not in original_modules and not module.startswith("google.protobuf") and not module.startswith("onnx") and not module.startswith("cv2"):
+ modules_to_remove.append(module)
+for module in modules_to_remove:
+ del sys.modules[module]
+
+# Restore original modules
+sys.modules.update(original_webui_modules)
|