Link to LitGPT for the Thunder example (#20)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Lightning-AI · Mar 21, 2024 · 164160f · 164160f
1 parent f0e57ed
commit 164160f
Show file tree

Hide file tree

Showing 12 changed files with 99 additions and 1,434 deletions.
diff --git a/README.md b/README.md
@@ -125,19 +125,7 @@ Thunder is in its early stages and should not be used for production runs yet.
 
 However, it can already deliver outstanding performance on LLM model supported by [LitGPT](https://github.com/Lightning-AI/lit-gpt), such as Mistral, Llama 2, Gemma, Falcon, and others.
 
-Run training loop for Llama, single-GPU:
-
-```bash
-python examples/lit-gpt/train.py
-```
-
-Run training loop for Llama, multi-GPU, using FSDP:
-
-```bash
-python examples/lit-gpt/train_fsdp.py
-```
-
-See [README.md](examples/lit-gpt/README.md) for details on running LitGPT with Thunder.
+Check out [the LitGPT integration](https://github.com/Lightning-AI/litgpt/tree/main/extensions/thunder) to learn about running LitGPT and Thunder together.
 
 ## Features
 

diff --git a/examples/lit-gpt/.gitignore b/examples/lit-gpt/.gitignore
diff --git a/examples/lit-gpt/1_forward.py b/examples/lit-gpt/1_forward.py
diff --git a/examples/lit-gpt/README.md b/examples/lit-gpt/README.md