diff --git a/requirements/notebooks.txt b/requirements/notebooks.txt index 54f29b2d2b..1eb53c6d15 100644 --- a/requirements/notebooks.txt +++ b/requirements/notebooks.txt @@ -5,4 +5,4 @@ ipython[all] ~=8.37.0 numpy liger-kernel == 0.4.0 cuda-python >=12.6.1, <14.0.0 -litgpt == 0.5.1 +litgpt == 0.5.11 diff --git a/requirements/test.txt b/requirements/test.txt index 28c5699ba2..cd7c182d88 100644 --- a/requirements/test.txt +++ b/requirements/test.txt @@ -12,7 +12,7 @@ expecttest ==0.3.0 # for test_ddp.py hypothesis ~=6.142.5 # for test_ddp.py numpy einops # for test_einops.py -litgpt==0.5.0 # for the model definition in tests and benchmarks # todo: need update to latest +litgpt==0.5.11 # for the model definition in tests and benchmarks # todo: need update to latest absl-py # thunder/benchmarks/test_benchmark_litgpt.py pandas # thunder/benchmarks/test_benchmark_litgpt.py xlsxwriter # thunder/benchmarks/test_benchmark_litgpt.py