From 3a2b27e9b38e639133913c42c03acdf0e136ec4f Mon Sep 17 00:00:00 2001
From: Masaki Kozuki <mkozuki@nvidia.com>
Date: Fri, 26 Apr 2024 05:26:08 +0900
Subject: [PATCH] Install `litgpt` from pypi (#274)

---
 requirements/test.txt | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/requirements/test.txt b/requirements/test.txt
index bd29df5b05..733ec0474d 100644
--- a/requirements/test.txt
+++ b/requirements/test.txt
@@ -11,7 +11,7 @@ expecttest ==0.2.1  # for test_ddp.py
 hypothesis ==6.100.0  # for test_ddp.py
 numpy  # for test_ops.py
 einops  # for test_einops.py
-litgpt @ git+https://github.com/Lightning-AI/lit-gpt@54628ecbdf72b2ceb42bb0ef5f74f1a89ec13d2f
+litgpt==0.3.0  # for the model definition in tests and benchmarks
 absl-py # thunder/benchmarks/test_benchmark_litgpt.py
 pandas # thunder/benchmarks/test_benchmark_litgpt.py
 xlsxwriter # thunder/benchmarks/test_benchmark_litgpt.py