From 29a33cbd25271fc07be86a51049e81aba126e88f Mon Sep 17 00:00:00 2001
From: Sebastian Raschka <mail@sebastianraschka.com>
Date: Thu, 31 Oct 2024 11:22:22 -0500
Subject: [PATCH] 0.5.3 post release setup (#1817)

---
 pyproject.toml | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 5d00a76075..406f5f2e1a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "litgpt"
-version = "0.5.3"
+version = "0.5.4.dev1"
 description = "Hackable implementation of state-of-the-art open-source LLMs"
 authors = [
     { name = "Lightning AI", email = "contact@lightning.ai" },
@@ -17,6 +17,7 @@ dependencies = [
     "safetensors>=0.4.3",               # download models
     "tokenizers>=0.15.2",               # tokenization in most models
     "tqdm>=4.66.0",                     # convert_hf_checkpoint
+    "lightning-thunder @ git+https://github.com/Lightning-AI/lightning-thunder/ ; python_version >= '3.10' and sys_platform == 'linux'",
 ]
 
 [project.urls]