diff --git a/setup.py b/setup.py index 9bffd60de..ef33b122a 100644 --- a/setup.py +++ b/setup.py @@ -61,7 +61,7 @@ "accelerate>=0.20.3", "pynvml==11.5.3", "compressed-tensors" - if version_info.is_release + if version_info.build_type == "release" else "compressed-tensors-nightly", ], extras_require={ diff --git a/src/llmcompressor/version.py b/src/llmcompressor/version.py index f3d279a95..3d97aaddf 100644 --- a/src/llmcompressor/version.py +++ b/src/llmcompressor/version.py @@ -8,7 +8,7 @@ from typing import Optional, Tuple # Define the base version and build type -version_base = "0.1.0" +version_base = "0.2.0" build_type = "dev" # can be 'release', 'nightly', 'dev', or 'dev' with a dev number