diff --git a/setup.py b/setup.py index f6a3d30f7..f42c8d38c 100644 --- a/setup.py +++ b/setup.py @@ -60,7 +60,7 @@ "datasets", "accelerate>=0.20.3,!=1.1.0", "pynvml", - "compressed-tensors==0.9.2" + "compressed-tensors" if version_info.build_type == "release" else "compressed-tensors-nightly", ],