diff --git a/pyproject.toml b/pyproject.toml index 43143c1f..87bc98d1 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -44,7 +44,7 @@ dependencies = [ "spacy", "tf-keras", "torch<2.9", - "pytorch-lightning<=2.5.6", + "pytorch-lightning<=2.6.0", "sentence-transformers<5.2", "timm", "torchvision",