diff --git a/pyproject.toml b/pyproject.toml index f11d8b510..20aefa782 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -5,7 +5,7 @@ readme = "README.md" license = {file = "LICENSE"} requires-python = "==3.10.*" dependencies = [ - "torch==2.7.1", + "torch==2.8.0", "tqdm>=4.65.0,<5.0.0", "transformers==4.56.1", "numpy>=1.26.0, <2.0.0", diff --git a/requirements.txt b/requirements.txt index 0fc428bee..00f76f9f0 100644 --- a/requirements.txt +++ b/requirements.txt @@ -545,7 +545,7 @@ toml==0.10.2 # via h2o-llmstudio tomli==2.3.0 # via h2o-cloud-discovery -torch==2.7.1+cu126 +torch==2.8.0 # via # accelerate # bitsandbytes