diff --git a/requirements/coverage.txt b/requirements/coverage.txt index f76b5b04e0..becc21ea9a 100644 --- a/requirements/coverage.txt +++ b/requirements/coverage.txt @@ -6,4 +6,4 @@ transformers ==4.52.4 lightning_sdk diffusers==0.35.1 accelerate -bitsandbytes==0.48.0 +bitsandbytes==0.48.2 diff --git a/requirements/test.txt b/requirements/test.txt index 28c5699ba2..9b367bee2b 100644 --- a/requirements/test.txt +++ b/requirements/test.txt @@ -17,7 +17,7 @@ absl-py # thunder/benchmarks/test_benchmark_litgpt.py pandas # thunder/benchmarks/test_benchmark_litgpt.py xlsxwriter # thunder/benchmarks/test_benchmark_litgpt.py jsonargparse # thunder/benchmarks/benchmark_litgpt.py -bitsandbytes==0.48.0; 'arm' not in platform_machine and 'aarch' not in platform_machine +bitsandbytes==0.48.2; 'arm' not in platform_machine and 'aarch' not in platform_machine bitsandbytes>=0.42,<0.43; 'arm' in platform_machine or 'aarch' in platform_machine transformers==4.55.4 # for test_networks.py diffusers==0.35.1 # for test_networks.py