diff --git a/setup.py b/setup.py index f42c8d38c..f6a3d30f7 100644 --- a/setup.py +++ b/setup.py @@ -60,7 +60,7 @@ "datasets", "accelerate>=0.20.3,!=1.1.0", "pynvml", - "compressed-tensors" + "compressed-tensors==0.9.2" if version_info.build_type == "release" else "compressed-tensors-nightly", ], diff --git a/src/llmcompressor/version.py b/src/llmcompressor/version.py index a91e351d2..de64a34be 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.4.0" +version_base = "0.4.1" build_type = "dev" # can be 'release', 'nightly', 'dev', or 'dev' with a dev number