diff --git a/comfy_kitchen/__init__.py b/comfy_kitchen/__init__.py index 55c611a..0467b51 100644 --- a/comfy_kitchen/__init__.py +++ b/comfy_kitchen/__init__.py @@ -16,7 +16,12 @@ # Import registry and exceptions from .registry import registry -__version__ = "0.1.0" +try: + from importlib.metadata import version, PackageNotFoundError + + __version__ = version("comfy-kitchen") +except PackageNotFoundError: + __version__ = "0.0.0" # Fallback for editable/dev installs __all__ = [ # Exceptions diff --git a/comfy_kitchen/backends/cuda/dlpack_bindings.cpp b/comfy_kitchen/backends/cuda/dlpack_bindings.cpp index 12ac4b4..fa9a9d4 100644 --- a/comfy_kitchen/backends/cuda/dlpack_bindings.cpp +++ b/comfy_kitchen/backends/cuda/dlpack_bindings.cpp @@ -553,8 +553,6 @@ NB_MODULE(_C, m) { // Feature availability flag (computed at module load time) m.attr("HAS_CUBLASLT") = comfy::CublasLtRuntime::instance().is_available(); - // Add version info - m.attr("__version__") = "0.1.0"; m.attr("__nanobind__") = true; m.attr("__stable_abi__") = true; }