diff --git a/pyproject.toml b/pyproject.toml
index f5e4ac8c44..67e96d4e76 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -27,7 +27,7 @@ dependencies = [
     "omegaconf",
 
     # Quantization
-    "torchao",  # ==0.3.1",
+    "torchao==0.3.1",
 ]
 dynamic = ["version"]