diff --git a/pyproject.toml b/pyproject.toml index ab7b9b26f3..40029ae13b 100644 --- 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