From f0abc765ca489f1423461b9e2554e17a7b2e3d0f Mon Sep 17 00:00:00 2001 From: Dipika Sikka Date: Thu, 12 Dec 2024 12:30:18 -0500 Subject: [PATCH] Update setup.py --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index d91c47aa2..71a681b48 100644 --- a/setup.py +++ b/setup.py @@ -60,7 +60,7 @@ "datasets", "accelerate>=0.20.3,!=1.1.0", "pynvml==11.5.3", - "compressed-tensors==0.8.1" + "compressed-tensors" if version_info.build_type == "release" else "compressed-tensors-nightly", ],