Update thunder pin

Lightning-AI · May 6, 2024 · e9ea126 · e9ea126
1 parent 8861daf
commit e9ea126
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/extensions/thunder/README.md b/extensions/thunder/README.md
@@ -612,7 +612,7 @@ Nvidia driver version: 545.23.08
 pytorch-triton==3.0.0+45fff310c8
 torch==2.4.0.dev20240427+cu121
 lightning==2.3.0.dev20240328
-git+https://github.com/Lightning-AI/lightning-thunder.git@ba75ad8d1578e38d2386cebbcc3cd8775d398df1#egg=lightning_thunder
+lightning-thunder==0.2.0.dev20240505
 nvfuser_cu121==0.2.3.dev20240428
 ```
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -30,7 +30,7 @@ test = [
     "transformers>=4.38.0",  # numerical comparisons
     "einops>=0.7.0",
     "protobuf>=4.23.4",
-    "lightning-thunder==0.2.0.dev20240404; python_version >= '3.10'",
+    "lightning-thunder==0.2.0.dev20240505; python_version >= '3.10'",
 ]
 all = [
     "bitsandbytes==0.42.0",      # quantization