diff --git a/setup.py b/setup.py index 1228b28..c5047bd 100644 --- a/setup.py +++ b/setup.py @@ -16,7 +16,7 @@ "zstandard==0.23.0", "pydantic-settings==2.8.1", "ray[serve]==2.44.0", - "torch==2.6.0", + "torch==2.7.1", "torchvision==0.21.0", "torchaudio==2.6.0", "transformers==4.50.0",