From f5e9d2b5313964b059a17f7a50410a9ca8fa8ea0 Mon Sep 17 00:00:00 2001
From: Jirka Borovec <6035284+Borda@users.noreply.github.com>
Date: Sun, 24 Mar 2024 21:55:06 +0100
Subject: [PATCH] bump litGPT to `940ffc96f7214bca24aa77479bc7c33900aaef28`
 (#66)

---
 requirements/notebooks.txt | 2 +-
 requirements/test.txt      | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/requirements/notebooks.txt b/requirements/notebooks.txt
index 7c37419cd6..c60affa8b3 100644
--- a/requirements/notebooks.txt
+++ b/requirements/notebooks.txt
@@ -1,3 +1,3 @@
 ipython[all] ==8.22.2
 
-litgpt @ git+https://github.com/Lightning-AI/lit-gpt@24d5eba1724c953b7506edc041a7da1ce226c129
+litgpt @ git+https://github.com/Lightning-AI/lit-gpt@940ffc96f7214bca24aa77479bc7c33900aaef28
diff --git a/requirements/test.txt b/requirements/test.txt
index 621ebaa7c5..36afeba5f5 100644
--- a/requirements/test.txt
+++ b/requirements/test.txt
@@ -11,7 +11,7 @@ expecttest ==0.2.1  # for test_ddp.py
 hypothesis ==6.99.10  # for test_ddp.py
 numpy  # for test_ops.py
 einops  # for test_einops.py
-litgpt @ git+https://github.com/Lightning-AI/lit-gpt@24d5eba1724c953b7506edc041a7da1ce226c129
+litgpt @ git+https://github.com/Lightning-AI/lit-gpt@940ffc96f7214bca24aa77479bc7c33900aaef28
 absl-py # thunder/benchmarks/test_benchmark_litgpt.py
 pandas # thunder/benchmarks/test_benchmark_litgpt.py
 xlsxwriter # thunder/benchmarks/test_benchmark_litgpt.py