diff --git a/requirements.txt b/requirements.txt index 437a7470f7..0bada4a2d4 100644 --- a/requirements.txt +++ b/requirements.txt @@ -77,7 +77,7 @@ partial_json_parser apache-tvm-ffi==0.1.6 # used for reduce nvidia-cutlass-dsl host overhead torch-c-dlpack-ext==0.1.3 # used for reduce nvidia-cutlass-dsl host overhead, optional package for improved torch tensor calling perf mistral-common==1.8.6 -torchao>=0.14.1 +torchao>=0.14.1,<0.16.0 cuda-core llist cuda-tile>=1.0.1