diff --git a/requirements/notebooks.txt b/requirements/notebooks.txt index 54f29b2d2b..5c274352f0 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.10 diff --git a/requirements/test.txt b/requirements/test.txt index f8dd57505d..d38d2f953e 100644 --- a/requirements/test.txt +++ b/requirements/test.txt @@ -12,7 +12,7 @@ expecttest ==0.3.0 # for test_ddp.py hypothesis ~=6.136.6 # 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.10 # 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