fix version__ (#122)

microsoft · Nov 18, 2024 · 27b70af · 27b70af
1 parent c5d86da
commit 27b70af
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "vptq"
-version = "0.0.3"
+version = "0.0.4"
 authors = [
   { name="Yang Wang", email="wyatuestc@gmai.com" },
   { name="Jicheng Wen", email="wejoincy@gmail.com"},

diff --git a/vptq/__init__.py b/vptq/__init__.py
@@ -3,7 +3,10 @@
 # Licensed under the MIT License.
 # --------------------------------------------------------------------------
 
-__version__ = "0.0.4"
+import importlib.metadata
+
 from vptq.layers import AutoModelForCausalLM, VQuantLinear
 
+__version__ = importlib.metadata.version("vptq")
+
 __all__ = ["AutoModelForCausalLM", "VQuantLinear"]