only install torchao for torch versions >= 2.4.0 (#1963)

2024-10-12 20:53:48 -04:00
parent 09bf1ceacc
commit d20b48a61e
2 changed files with 7 additions and 1 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -52,3 +52,5 @@ lm_eval==0.4.4
 langdetect==1.0.9
 immutabledict==4.2.0
 antlr4-python3-runtime==4.13.2
+
+torchao==0.5.0