comfy-env 0.0.67__py3-none-any.whl → 0.0.68__py3-none-any.whl

comfy_env/pixi/core.py

@@ -252,24 +252,10 @@ def pixi_install(
     dependencies.setdefault("pip", "*")  # Always include pip
     pixi_data["dependencies"] = dependencies
 
-    # Add pypi-options for CUDA wheels
+    # Add pypi-options for PyTorch index (CUDA packages installed separately via pip)
     if cfg.has_cuda and cuda_version:
         pypi_options = pixi_data.get("pypi-options", {})
-        # Merge find-links (pixi expects [{url: "..."}, ...] format)
-        find_links = pypi_options.get("find-links", [])
-        existing_urls = {
-            entry.get("url") if isinstance(entry, dict) else entry
-            for entry in find_links
-        }
-        if CUDA_WHEELS_INDEX not in existing_urls:
-            find_links.append({"url": CUDA_WHEELS_INDEX})
-        # Normalize any plain strings to {url: ...} format
-        find_links = [
-            {"url": entry} if isinstance(entry, str) else entry
-            for entry in find_links
-        ]
-        pypi_options["find-links"] = find_links
-        # Merge extra-index-urls
+        # Add PyTorch CUDA index for torch installation
         cuda_short = cuda_version.replace(".", "")[:3]
         pytorch_index = f"https://download.pytorch.org/whl/cu{cuda_short}"
         extra_urls = pypi_options.get("extra-index-urls", [])
@@ -311,23 +297,31 @@ def pixi_install(
         log(f"pixi install failed:\n{result.stderr}")
         raise RuntimeError(f"pixi install failed: {result.stderr}")
 
-    # Install CUDA packages with --no-deps (avoids PyPI version conflicts)
+    # Install CUDA packages with --no-index --find-links (bypasses PyPI completely)
     if cfg.cuda_packages and cuda_version:
-        log(f"Installing CUDA packages with --no-deps: {cfg.cuda_packages}")
+        log(f"Installing CUDA packages: {cfg.cuda_packages}")
         python_path = get_pixi_python(node_dir)
         if not python_path:
             raise RuntimeError("Could not find Python in pixi environment")
 
-        pip_cmd = [
-            str(python_path), "-m", "pip", "install",
-            "--no-deps",
-            "--index-url", CUDA_WHEELS_INDEX,
-        ] + cfg.cuda_packages
+        for package in cfg.cuda_packages:
+            # Each package has its own find-links page at CUDA_WHEELS_INDEX/<package>/
+            find_links_url = f"{CUDA_WHEELS_INDEX}{package}/"
+            log(f"  Installing {package} from {find_links_url}")
+
+            pip_cmd = [
+                str(python_path), "-m", "pip", "install",
+                "--no-index",
+                "--no-deps",
+                "--no-cache-dir",
+                "--find-links", find_links_url,
+                package,
+            ]
+            result = subprocess.run(pip_cmd, capture_output=True, text=True)
+            if result.returncode != 0:
+                log(f"CUDA package install failed for {package}:\n{result.stderr}")
+                raise RuntimeError(f"CUDA package install failed: {result.stderr}")
 
-        result = subprocess.run(pip_cmd, capture_output=True, text=True)
-        if result.returncode != 0:
-            log(f"CUDA package install failed:\n{result.stderr}")
-            raise RuntimeError(f"CUDA package install failed: {result.stderr}")
         log("CUDA packages installed")
 
     # Create symlink/junction to _env_<name> for discovery (only for isolated subdirs)

comfy_env/workers/mp.py

@@ -430,20 +430,68 @@ class MPWorker(Worker):
             return
 
         # Import torch here to avoid import at module level
+        import os
+        import sys
+
+        # Clear conda/pixi environment variables FIRST, before importing multiprocessing
+        # These can cause the child process to pick up the wrong Python interpreter
+        # or stdlib, leading to sys.version mismatch errors in platform module
+        conda_env_vars = [
+            'CONDA_PREFIX',
+            'CONDA_DEFAULT_ENV',
+            'CONDA_PYTHON_EXE',
+            'CONDA_EXE',
+            'CONDA_SHLVL',
+            'PYTHONHOME',
+            'PYTHONPATH',  # Also clear PYTHONPATH to prevent pixi paths
+            '_CE_CONDA',
+            '_CE_M',
+        ]
+        saved_env = {}
+        for var in conda_env_vars:
+            if var in os.environ:
+                saved_env[var] = os.environ.pop(var)
+
+        # Also remove pixi paths from LD_LIBRARY_PATH
+        ld_lib = os.environ.get('LD_LIBRARY_PATH', '')
+        if '.pixi' in ld_lib:
+            saved_env['LD_LIBRARY_PATH'] = ld_lib
+            # Filter out pixi paths
+            new_ld_lib = ':'.join(p for p in ld_lib.split(':') if '.pixi' not in p)
+            if new_ld_lib:
+                os.environ['LD_LIBRARY_PATH'] = new_ld_lib
+            else:
+                os.environ.pop('LD_LIBRARY_PATH', None)
+
         import torch.multiprocessing as mp
 
-        # Use spawn to get clean subprocess (no inherited CUDA context)
-        ctx = mp.get_context('spawn')
-
-        self._queue_in = ctx.Queue()
-        self._queue_out = ctx.Queue()
-        self._process = ctx.Process(
-            target=_worker_loop,
-            args=(self._queue_in, self._queue_out, self._sys_path, self._lib_path),
-            daemon=True,
-        )
-        self._process.start()
-        self._started = True
+        try:
+            # Use spawn to get clean subprocess (no inherited CUDA context)
+            ctx = mp.get_context('spawn')
+
+            # Explicitly set the spawn executable to the current Python
+            # This prevents pixi/conda from hijacking the spawn process
+            import multiprocessing.spawn as mp_spawn
+            original_exe = mp_spawn.get_executable()
+            if original_exe != sys.executable.encode() and original_exe != sys.executable:
+                print(f"[comfy-env] Warning: spawn executable was {original_exe}, forcing to {sys.executable}")
+            mp_spawn.set_executable(sys.executable)
+
+            self._queue_in = ctx.Queue()
+            self._queue_out = ctx.Queue()
+            self._process = ctx.Process(
+                target=_worker_loop,
+                args=(self._queue_in, self._queue_out, self._sys_path, self._lib_path),
+                daemon=True,
+            )
+            self._process.start()
+            self._started = True
+
+            # Restore original executable setting
+            mp_spawn.set_executable(original_exe)
+        finally:
+            # Restore env vars in parent process
+            os.environ.update(saved_env)
 
     def call(
         self,

{comfy_env-0.0.67.dist-info → comfy_env-0.0.68.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: comfy-env
-Version: 0.0.67
+Version: 0.0.68
 Summary: Environment management for ComfyUI custom nodes - CUDA wheel resolution and process isolation
 Project-URL: Homepage, https://github.com/PozzettiAndrea/comfy-env
 Project-URL: Repository, https://github.com/PozzettiAndrea/comfy-env

{comfy_env-0.0.67.dist-info → comfy_env-0.0.68.dist-info}/RECORD

@@ -10,7 +10,7 @@ comfy_env/config/types.py,sha256=H0m_L4nURv-9M-JZvTcyG9VEcbWzBlppA3Xin26B0jM,188
 comfy_env/isolation/__init__.py,sha256=vw9a4mpJ2CFjy-PLe_A3zQ6umBQklgqWNxwn9beNw3g,175
 comfy_env/isolation/wrap.py,sha256=9bXxK9h4FMrFL9k4EAdILWxkjXYytVCJV68MuwwufK8,11485
 comfy_env/pixi/__init__.py,sha256=BUrq7AQf3WDm0cHWh72B2xZbURNnDu2dCuELWiQCUiM,997
-comfy_env/pixi/core.py,sha256=iYTvCxmf2M9fxK9cM9RyKQyOSJVkpHyR7GXPm45ZFRI,12656
+comfy_env/pixi/core.py,sha256=_XXS8SEM9VHe3abDCK98Ky8zyLKYRsLBJfVazSNzuKs,12475
 comfy_env/pixi/cuda_detection.py,sha256=sqB3LjvGNdV4eFqiARQGfyecBM3ZiUmeh6nG0YCRYQw,9751
 comfy_env/pixi/resolver.py,sha256=U_A8rBDxCj4gUlJt2YJQniP4cCKqxJEiVFgXOoH7vM8,6339
 comfy_env/pixi/platform/__init__.py,sha256=Nb5MPZIEeanSMEWwqU4p4bnEKTJn1tWcwobnhq9x9IY,614
@@ -22,11 +22,11 @@ comfy_env/templates/comfy-env-instructions.txt,sha256=ve1RAthW7ouumU9h6DM7mIRX1M
 comfy_env/templates/comfy-env.toml,sha256=TBa_8V8gdwoRej4Lb_mkqZ_Ytk_quFTnbwwc1YK_gHE,4190
 comfy_env/workers/__init__.py,sha256=TMVG55d2XLP1mJ3x1d16H0SBDJZtk2kMC5P4HLk9TrA,1073
 comfy_env/workers/base.py,sha256=4ZYTaQ4J0kBHCoO_OfZnsowm4rJCoqinZUaOtgkOPbw,2307
-comfy_env/workers/mp.py,sha256=E_SUvvCd11uCW5dIURET8EVHcgRtu36MPOEjCFDq4Ok,23920
+comfy_env/workers/mp.py,sha256=d6PFVrgqp3MK7Gkt08a8LQD_VSwHtoIcGx2Lovou3vM,25972
 comfy_env/workers/subprocess.py,sha256=UMhKhaJoSjv2-FWnwQq9TeMWtaDLIs3ajAFTq1ngdJw,57844
 comfy_env/workers/tensor_utils.py,sha256=TCuOAjJymrSbkgfyvcKtQ_KbVWTqSwP9VH_bCaFLLq8,6409
-comfy_env-0.0.67.dist-info/METADATA,sha256=J0u--ky-c7j_vt0j8vu2F3tqZrYS8mi_wTqXjMJkx1M,6946
-comfy_env-0.0.67.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-comfy_env-0.0.67.dist-info/entry_points.txt,sha256=J4fXeqgxU_YenuW_Zxn_pEL7J-3R0--b6MS5t0QmAr0,49
-comfy_env-0.0.67.dist-info/licenses/LICENSE,sha256=E68QZMMpW4P2YKstTZ3QU54HRQO8ecew09XZ4_Vn870,1093
-comfy_env-0.0.67.dist-info/RECORD,,
+comfy_env-0.0.68.dist-info/METADATA,sha256=iIsLoiV1bqVdk17NyzD4TOYnStx2c290dDhp0aZosB4,6946
+comfy_env-0.0.68.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+comfy_env-0.0.68.dist-info/entry_points.txt,sha256=J4fXeqgxU_YenuW_Zxn_pEL7J-3R0--b6MS5t0QmAr0,49
+comfy_env-0.0.68.dist-info/licenses/LICENSE,sha256=E68QZMMpW4P2YKstTZ3QU54HRQO8ecew09XZ4_Vn870,1093
+comfy_env-0.0.68.dist-info/RECORD,,