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

comfy_env/wheel_sources.yml

@@ -1,141 +0,0 @@
-# Wheel sources registry for CUDA packages
-#
-# Each package has a single wheel_template that resolves to a direct .whl URL.
-# Users can override these in their comfy-env.toml [wheel_sources] section.
-#
-# Template variables:
-#   {version}        - Package version (e.g., "0.4.0")
-#   {cuda_version}   - Full CUDA version (e.g., "12.8")
-#   {cuda_short}     - CUDA without dot (e.g., "128")
-#   {cuda_major}     - CUDA major version (e.g., "12")
-#   {torch_version}  - Full PyTorch version (e.g., "2.8.0")
-#   {torch_short}    - PyTorch without dots (e.g., "280")
-#   {torch_mm}       - PyTorch major.minor no dot (e.g., "28")
-#   {torch_dotted_mm}- PyTorch major.minor with dot (e.g., "2.8")
-#   {py_version}     - Python version (e.g., "3.10")
-#   {py_short}       - Python without dot (e.g., "310")
-#   {py_minor}       - Python minor version only (e.g., "10")
-#   {py_tag}         - Python tag (e.g., "cp310")
-#   {platform}       - Platform tag (e.g., "linux_x86_64")
-
-packages:
-  # ===========================================================================
-  # PyTorch Geometric (PyG) - official wheels
-  # https://pytorch-geometric.readthedocs.io/en/latest/install/installation.html
-  # ===========================================================================
-  torch-scatter:
-    wheel_template: "https://data.pyg.org/whl/torch-{torch_version}%2Bcu{cuda_short}/torch_scatter-{version}%2Bpt{torch_mm}cu{cuda_short}-{py_tag}-{py_tag}-{platform}.whl"
-    default_version: "2.1.2"
-    description: Scatter operations for PyTorch
-
-  torch-cluster:
-    wheel_template: "https://data.pyg.org/whl/torch-{torch_version}%2Bcu{cuda_short}/torch_cluster-{version}%2Bpt{torch_mm}cu{cuda_short}-{py_tag}-{py_tag}-{platform}.whl"
-    default_version: "1.6.3"
-    description: Clustering algorithms for PyTorch
-
-  torch-sparse:
-    wheel_template: "https://data.pyg.org/whl/torch-{torch_version}%2Bcu{cuda_short}/torch_sparse-{version}%2Bpt{torch_mm}cu{cuda_short}-{py_tag}-{py_tag}-{platform}.whl"
-    default_version: "0.6.18"
-    description: Sparse tensor operations for PyTorch
-
-  torch-spline-conv:
-    wheel_template: "https://data.pyg.org/whl/torch-{torch_version}%2Bcu{cuda_short}/torch_spline_conv-{version}%2Bpt{torch_mm}cu{cuda_short}-{py_tag}-{py_tag}-{platform}.whl"
-    default_version: "1.2.2"
-    description: Spline convolutions for PyTorch
-
-  # ===========================================================================
-  # PozzettiAndrea cuda-wheels (GitHub releases)
-  # https://github.com/PozzettiAndrea/cuda-wheels/releases
-  # ===========================================================================
-  pytorch3d:
-    wheel_template: "https://github.com/PozzettiAndrea/cuda-wheels/releases/download/pytorch3d-latest/pytorch3d-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
-    default_version: "0.7.9"
-    description: PyTorch3D - 3D deep learning library
-
-  nvdiffrast:
-    wheel_template: "https://github.com/PozzettiAndrea/cuda-wheels/releases/download/nvdiffrast-latest/nvdiffrast-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
-    default_version: "0.4.0"
-    description: NVIDIA differentiable rasterizer
-
-  cumesh:
-    wheel_template: "https://github.com/PozzettiAndrea/cuda-wheels/releases/download/cumesh-latest/cumesh-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
-    default_version: "0.0.1"
-    description: CUDA-accelerated mesh utilities
-
-  cubvh:
-    wheel_template: "https://github.com/PozzettiAndrea/cuda-wheels/releases/download/cubvh-latest/cubvh-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
-    default_version: "0.1.2"
-    description: CUDA BVH acceleration for mesh operations and marching cubes
-
-  o_voxel:
-    wheel_template: "https://github.com/PozzettiAndrea/cuda-wheels/releases/download/o_voxel-latest/o_voxel-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
-    default_version: "0.0.1"
-    description: O-Voxel CUDA extension for TRELLIS
-
-  flex_gemm:
-    wheel_template: "https://github.com/PozzettiAndrea/cuda-wheels/releases/download/flex_gemm-latest/flex_gemm-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
-    default_version: "1.0.0"
-    description: Flexible GEMM operations
-
-  nvdiffrec_render:
-    wheel_template: "https://github.com/PozzettiAndrea/cuda-wheels/releases/download/nvdiffrec_render-latest/nvdiffrec_render-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
-    default_version: "0.0.1"
-    description: NVDiffRec rendering utilities
-
-  gsplat:
-    wheel_template: "https://github.com/PozzettiAndrea/cuda-wheels/releases/download/gsplat-latest/gsplat-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
-    default_version: "1.5.3"
-    description: Gaussian splatting rasterization
-
-  cc_torch:
-    wheel_template: "https://github.com/PozzettiAndrea/cuda-wheels/releases/download/cc_torch-latest/cc_torch-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
-    default_version: "0.2"
-    description: GPU-accelerated connected components
-
-  torch_generic_nms:
-    wheel_template: "https://github.com/PozzettiAndrea/cuda-wheels/releases/download/torch_generic_nms-latest/torch_generic_nms-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
-    default_version: "0.1"
-    description: GPU-accelerated Non-Maximum Suppression
-
-  lietorch:
-    wheel_template: "https://github.com/PozzettiAndrea/cuda-wheels/releases/download/lietorch-latest/lietorch-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
-    default_version: "0.3"
-    description: Lie group operations for PyTorch (DPVO dependency)
-
-  sageattention:
-    wheel_template: "https://github.com/PozzettiAndrea/cuda-wheels/releases/download/sageattention-latest/sageattention-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
-    default_version: "2.2.0"
-    description: SageAttention - 2-5x faster than FlashAttention with quantized kernels
-
-  dpvo-cuda:
-    wheel_template: "https://github.com/PozzettiAndrea/cuda-wheels/releases/download/dpvo_cuda-latest/dpvo_cuda-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
-    default_version: "0.0.0"
-    description: DPVO CUDA extensions (cuda_corr, cuda_ba, lietorch_backends) - torch 2.4 only
-
-  # ===========================================================================
-  # detectron2 - Facebook's detection library
-  # https://github.com/facebookresearch/detectron2
-  # Prebuilt wheels from miropsota's torch_packages_builder
-  # ===========================================================================
-  detectron2:
-    find_links: "https://miropsota.github.io/torch_packages_builder/detectron2/"
-    default_version: "0.6"
-    description: Detectron2 - Facebook's detection and segmentation library
-
-  # ===========================================================================
-  # flash-attn - Dao-AILab (Linux x86_64)
-  # For other platforms, users should add custom wheel_source in their config
-  # ===========================================================================
-  flash-attn:
-    wheel_template: "https://github.com/Dao-AILab/flash-attention/releases/download/v{version}/flash_attn-{version}%2Bcu{cuda_major}torch{torch_dotted_mm}cxx11abiTRUE-{py_tag}-{py_tag}-linux_x86_64.whl"
-    description: Flash Attention for fast transformer inference (Linux x86_64)
-
-  # ===========================================================================
-  # spconv - Sparse convolution library
-  # PyPI with CUDA-versioned package names (spconv-cu124, spconv-cu126, etc.)
-  # Note: This uses a special package_name field, not wheel_template
-  # ===========================================================================
-  spconv:
-    wheel_template: "https://github.com/PozzettiAndrea/cuda-wheels/releases/download/spconv_cu{cuda_short}-latest/spconv_cu{cuda_short}-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
-    default_version: "2.3.8"
-    description: Sparse convolution library

comfy_env/workers/pool.py

@@ -1,241 +0,0 @@
-"""
-WorkerPool - Global registry and management of named workers.
-
-Provides a simple API for getting workers by name:
-
-    from comfy_env.workers import get_worker
-
-    worker = get_worker("sam3d")
-    result = worker.call_module("my_module", "my_func", image=tensor)
-
-Workers are registered at startup and reused across calls:
-
-    from comfy_env.workers import register_worker, TorchMPWorker
-
-    register_worker("default", TorchMPWorker())
-    register_worker("sam3d", PersistentVenvWorker(
-        python="/path/to/venv/bin/python",
-        working_dir="/path/to/nodes",
-    ))
-"""
-
-import atexit
-import threading
-from typing import Dict, Optional, Union
-from pathlib import Path
-
-from .base import Worker
-
-
-class WorkerPool:
-    """
-    Singleton pool of named workers.
-
-    Manages worker lifecycle, provides access by name, handles cleanup.
-    """
-
-    _instance: Optional["WorkerPool"] = None
-    _lock = threading.Lock()
-
-    def __new__(cls):
-        if cls._instance is None:
-            with cls._lock:
-                if cls._instance is None:
-                    cls._instance = super().__new__(cls)
-                    cls._instance._initialized = False
-        return cls._instance
-
-    def __init__(self):
-        if self._initialized:
-            return
-        self._initialized = True
-        self._workers: Dict[str, Worker] = {}
-        self._factories: Dict[str, callable] = {}
-        self._worker_lock = threading.Lock()
-
-    def register(
-        self,
-        name: str,
-        worker: Optional[Worker] = None,
-        factory: Optional[callable] = None,
-    ) -> None:
-        """
-        Register a worker or worker factory.
-
-        Args:
-            name: Name to register under.
-            worker: Pre-created worker instance.
-            factory: Callable that creates worker on first use (lazy).
-
-        Only one of worker or factory should be provided.
-        """
-        if worker is not None and factory is not None:
-            raise ValueError("Provide either worker or factory, not both")
-        if worker is None and factory is None:
-            raise ValueError("Must provide worker or factory")
-
-        with self._worker_lock:
-            # Shutdown existing worker if replacing
-            if name in self._workers:
-                try:
-                    self._workers[name].shutdown()
-                except:
-                    pass
-
-            if worker is not None:
-                self._workers[name] = worker
-                self._factories.pop(name, None)
-            else:
-                self._factories[name] = factory
-                self._workers.pop(name, None)
-
-    def get(self, name: str) -> Worker:
-        """
-        Get a worker by name.
-
-        Args:
-            name: Registered worker name.
-
-        Returns:
-            The worker instance.
-
-        Raises:
-            KeyError: If no worker registered with that name.
-        """
-        with self._worker_lock:
-            # Check for existing worker
-            if name in self._workers:
-                worker = self._workers[name]
-                if worker.is_alive():
-                    return worker
-                # Worker died, try to recreate from factory
-                if name not in self._factories:
-                    raise RuntimeError(f"Worker '{name}' died and no factory to recreate")
-
-            # Create from factory
-            if name in self._factories:
-                worker = self._factories[name]()
-                self._workers[name] = worker
-                return worker
-
-            raise KeyError(f"No worker registered with name: {name}")
-
-    def shutdown(self, name: Optional[str] = None) -> None:
-        """
-        Shutdown workers.
-
-        Args:
-            name: If provided, shutdown only this worker.
-                  If None, shutdown all workers.
-        """
-        with self._worker_lock:
-            if name is not None:
-                if name in self._workers:
-                    try:
-                        self._workers[name].shutdown()
-                    except:
-                        pass
-                    del self._workers[name]
-            else:
-                for worker in self._workers.values():
-                    try:
-                        worker.shutdown()
-                    except:
-                        pass
-                self._workers.clear()
-
-    def list_workers(self) -> Dict[str, str]:
-        """
-        List all registered workers.
-
-        Returns:
-            Dict of name -> status string.
-        """
-        with self._worker_lock:
-            result = {}
-            for name, worker in self._workers.items():
-                status = "alive" if worker.is_alive() else "dead"
-                result[name] = f"{type(worker).__name__} ({status})"
-            for name in self._factories:
-                if name not in result:
-                    result[name] = f"factory (not started)"
-            return result
-
-
-# Global pool instance
-_pool = WorkerPool()
-
-
-def get_worker(name: str) -> Worker:
-    """
-    Get a worker by name from the global pool.
-
-    Args:
-        name: Registered worker name.
-
-    Returns:
-        Worker instance.
-
-    Example:
-        worker = get_worker("sam3d")
-        result = worker.call_module("my_module", "my_func", image=tensor)
-    """
-    return _pool.get(name)
-
-
-def register_worker(
-    name: str,
-    worker: Optional[Worker] = None,
-    factory: Optional[callable] = None,
-) -> None:
-    """
-    Register a worker in the global pool.
-
-    Args:
-        name: Name to register under.
-        worker: Pre-created worker instance.
-        factory: Callable that creates worker on demand.
-
-    Example:
-        # Register pre-created worker
-        register_worker("default", TorchMPWorker())
-
-        # Register factory for lazy creation
-        register_worker("sam3d", factory=lambda: PersistentVenvWorker(
-            python="/path/to/venv/bin/python",
-        ))
-    """
-    _pool.register(name, worker=worker, factory=factory)
-
-
-def shutdown_workers(name: Optional[str] = None) -> None:
-    """
-    Shutdown workers in the global pool.
-
-    Args:
-        name: If provided, shutdown only this worker.
-              If None, shutdown all workers.
-    """
-    _pool.shutdown(name)
-
-
-def list_workers() -> Dict[str, str]:
-    """
-    List all registered workers.
-
-    Returns:
-        Dict of name -> status description.
-    """
-    return _pool.list_workers()
-
-
-# Register default worker (TorchMPWorker) on import
-def _register_default():
-    from .torch_mp import TorchMPWorker
-    register_worker("default", factory=lambda: TorchMPWorker(name="default"))
-
-
-_register_default()
-
-# Cleanup on exit
-atexit.register(lambda: shutdown_workers())

comfy_env-0.0.65.dist-info/RECORD

@@ -1,48 +0,0 @@
-comfy_env/__init__.py,sha256=DvOCFCq-EuXZ_e6vsPCObgTPYSRLZqiG3w35SOD-e4E,4101
-comfy_env/cli.py,sha256=Pjzb-jsoH67lyHxmBFOWvasWX01eHFE_BEjUk1l-uEo,14509
-comfy_env/decorator.py,sha256=SL2duhj8EpQuK-TuEgNA9anTgQI9gWza5ikPlLXESjA,27018
-comfy_env/errors.py,sha256=q-C3vyrPa_kk_Ao8l17mIGfJiG2IR0hCFV0GFcNLmcI,9924
-comfy_env/install.py,sha256=nfzGifXVd87a6dnwY3GtktrXpS7fjWuWF-QV9hh2e94,15657
-comfy_env/isolation.py,sha256=a-q6mA9tzD27kZSZCTioSKlK-3kpRFKYA23V_75OmaI,9286
-comfy_env/nodes.py,sha256=CWUe35jU5SKk4ur-SddZePdqWgxJDlxGhpcJiu5pAK4,4354
-comfy_env/pixi.py,sha256=DnDvX3aMtwbo8XPJUt4JYADUyWcXGJ72PqcpN7Fkg_A,27440
-comfy_env/registry.py,sha256=w-QwvAPFlCrBYRAv4cXkp2zujQPZn8Fk5DUxKCtox8o,3430
-comfy_env/resolver.py,sha256=WoNIo2IfTR2RlEf_HQl66eAeMa2R2pmLof_UdK-0RNE,6714
-comfy_env/stub_imports.py,sha256=qOYxCJ8BbIfMHGBfqC5KGZSK6AK-iqmSjWi8ZCP8SAM,8425
-comfy_env/env/__init__.py,sha256=o8-k213v2dnK1t2qsZW1z-zl5tLX65aWcEj7AFdrHwk,1126
-comfy_env/env/config.py,sha256=-hrGRU4Q7rRV55Q73KkkfKe6dLLRLTtuOE7triAbYEk,7572
-comfy_env/env/config_file.py,sha256=MXear-3a2wjagRVpP869XIiFHwFrM_PNIH27SRPQbbU,24103
-comfy_env/env/cuda_gpu_detection.py,sha256=sqB3LjvGNdV4eFqiARQGfyecBM3ZiUmeh6nG0YCRYQw,9751
-comfy_env/env/manager.py,sha256=-qdbZDsbNfs70onVbC7mhKCzNsxYx3WmG7ttlBinhGI,23659
-comfy_env/env/security.py,sha256=dNSitAnfBNVdvxgBBntYw33AJaCs_S1MHb7KJhAVYzM,8171
-comfy_env/env/platform/__init__.py,sha256=Nb5MPZIEeanSMEWwqU4p4bnEKTJn1tWcwobnhq9x9IY,614
-comfy_env/env/platform/base.py,sha256=iS0ptTTVjXRwPU4qWUdvHI7jteuzxGSjWr5BUQ7hGiU,2453
-comfy_env/env/platform/darwin.py,sha256=HK3VkLT6DfesAnIXwx2IaUFHTBclF0xTQnC7azWY6Kc,1552
-comfy_env/env/platform/linux.py,sha256=xLp8FEbFqZLQrzIZBI9z3C4g23Ab1ASTHLsXDzsdCoA,2062
-comfy_env/env/platform/windows.py,sha256=FCOCgpzGzorY9-HueMlJUR8DxM2eH-cj9iZk6K026Is,10891
-comfy_env/ipc/__init__.py,sha256=pTjgJn5YJxLXmEvuKh3lkCEJQs-6W6_F01jfkFMUi0c,1375
-comfy_env/ipc/bridge.py,sha256=zcyN3xzV4WWBrBFNwCniPBR58dLCg46-k9TtyW5U000,16437
-comfy_env/ipc/protocol.py,sha256=gfWe5yEDUn4QWhcdWFcxn40GqxlW1Uf23j0edOzPPng,7951
-comfy_env/ipc/tensor.py,sha256=DyU28GymKkLPVwzZyKdm2Av222hdaycMgv3KdL5mtO0,12009
-comfy_env/ipc/torch_bridge.py,sha256=WzdwDJa3N_1fEl9OeZxikvMbwryO5P63o0WmEDpS18A,13206
-comfy_env/ipc/transport.py,sha256=XQlRcfQsd4nd909KIYnZKvsS3ksGpGjyVucn8jvmLIU,9698
-comfy_env/ipc/worker.py,sha256=oxTLF9xXrl8CRx_JVNBdkxZh35NuzfkdxhaUtUuXogs,6661
-comfy_env/stubs/__init__.py,sha256=jMeWEKY30y8QqYX9AUyuZbmm607erQTc4N7YaDoAH00,38
-comfy_env/stubs/folder_paths.py,sha256=K90J34EG6LD4eZP8YG-xMeBmqwpp_wA8E92DKMXd1GQ,2189
-comfy_env/stubs/comfy/__init__.py,sha256=-y4L6gX21vrI2V8MvNaMeHOcAn5kUNK3jUyLvtXRmJQ,173
-comfy_env/stubs/comfy/model_management.py,sha256=Khx8Qa3NutKPLTn9oSM3VLeATUOg1fe4QCjxdxXd6eE,1462
-comfy_env/stubs/comfy/utils.py,sha256=s3t_KLj_-w1Uj3A3iAy69wIk4Ggklojw5hsDNb69Pcc,776
-comfy_env/templates/comfy-env-instructions.txt,sha256=Q38Hb_YdN0a8rTxn7l5ON4JDPba7XgVftDqfEy-8I2I,3004
-comfy_env/templates/comfy-env.toml,sha256=AcvPtlPlpqmwbz7vYy6SZCZOm3pL0HHXhsZwpAYXRAU,6466
-comfy_env/workers/__init__.py,sha256=IKZwOvrWOGqBLDUIFAalg4CdqzJ_YnAdxo2Ha7gZTJ0,1467
-comfy_env/workers/base.py,sha256=ZILYXlvGCWuCZXmjKqfG8VeD19ihdYaASdlbasl2BMo,2312
-comfy_env/workers/pool.py,sha256=MtjeOWfvHSCockq8j1gfnxIl-t01GSB79T5N4YB82Lg,6956
-comfy_env/workers/tensor_utils.py,sha256=TCuOAjJymrSbkgfyvcKtQ_KbVWTqSwP9VH_bCaFLLq8,6409
-comfy_env/workers/torch_mp.py,sha256=TnsCoBHEJBXEoBkx7WiCd9tBAlzFtMOw1dk_7_zGJZY,22288
-comfy_env/workers/venv.py,sha256=WdK8GZkLXyzkCI0Xdber_BJhJRP5eXMF0M2yd8GU-Zk,59809
-comfy_env/wheel_sources.yml,sha256=uU0YJmWaiLAicQNN9VYS8PZevlP2NOH6mBUE294dvAo,8156
-comfy_env-0.0.65.dist-info/METADATA,sha256=PGpxA6Rue44EzyH8WRMmwbpOW0OFZJVwT3cnPcDvKjc,8735
-comfy_env-0.0.65.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-comfy_env-0.0.65.dist-info/entry_points.txt,sha256=J4fXeqgxU_YenuW_Zxn_pEL7J-3R0--b6MS5t0QmAr0,49
-comfy_env-0.0.65.dist-info/licenses/LICENSE,sha256=E68QZMMpW4P2YKstTZ3QU54HRQO8ecew09XZ4_Vn870,1093
-comfy_env-0.0.65.dist-info/RECORD,,