diff --git a/README.md b/README.md index fea0a40..ff42dd4 100644 --- a/README.md +++ b/README.md @@ -43,12 +43,33 @@ A stereoscopic 3D toolkit for ComfyUI that combines three solutions into one uni ```bash cd ComfyUI/custom_nodes/ git clone https://github.com/Dobidop/ComfyStereo.git +cd ComfyStereo ``` -2. Install core dependencies: +2. Install dependencies (choose one): + +**Option A: Using requirements.txt (traditional)** ```bash -cd ComfyStereo +# Base only (stereo generation + DeoVR) pip install -r requirements.txt + +# Or install all features at once +pip install -r requirements.txt +``` + +**Option B: Using pip with optional dependencies (recommended)** +```bash +# Base only +pip install -e . + +# Base + VR viewing +pip install -e ".[vr]" + +# Base + AI stereo generation +pip install -e ".[ai]" + +# Everything (all features) +pip install -e ".[all]" ``` 3. Restart ComfyUI diff --git a/pyproject.toml b/pyproject.toml index 1eeb8f3..172305d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,10 +1,15 @@ [project] name = "comfystereo" -description = "A simple node for stereoscopic image generation. Nodes: Stereo Image Node - a basic port from the Automatic1111 stereo script with som extra features added, such as edge and direction aware blurring of the depth map. DeoVR View - A node for launching and viewing images and videos in DeoVR. See readme for more details and setup of autostart of DeoVR" -version = "1.2.2" +description = "Comprehensive stereoscopic 3D toolkit for ComfyUI combining depth-based stereo generation with GPU acceleration, native VR viewing via PyOpenXR, and AI-powered StereoDiffusion. Supports 6+ nodes for stereo conversion, VR headset viewing, and diffusion model stereo generation." +version = "2.0.0" license = {file = "LICENSE"} dependencies = ["torch", "numpy", "Pillow", "opencv-python", "numba", "scipy", "psutil"] +[project.optional-dependencies] +vr = ["pyopenxr>=1.0.0", "PyOpenGL>=3.1.0", "PyOpenGL_accelerate>=3.1.0", "glfw>=2.0.0", "pygame>=2.0.0"] +ai = ["diffusers>=0.21.0", "transformers", "accelerate", "einops", "tqdm", "scikit-image"] +all = ["pyopenxr>=1.0.0", "PyOpenGL>=3.1.0", "PyOpenGL_accelerate>=3.1.0", "glfw>=2.0.0", "pygame>=2.0.0", "diffusers>=0.21.0", "transformers", "accelerate", "einops", "tqdm", "scikit-image"] + [project.urls] Repository = "https://github.com/Dobidop/ComfyStereo" # Used by Comfy Registry https://comfyregistry.org