diff --git a/TabPFN/pyproject.toml b/TabPFN/pyproject.toml index b25d281..a4e59f7 100644 --- a/TabPFN/pyproject.toml +++ b/TabPFN/pyproject.toml @@ -15,7 +15,7 @@ description = "Interface for using TabPFN and library to train TabPFN'" readme = "README.md" requires-python = ">=3.7" dependencies=[ - 'torch>=1.9.0', + 'torch>=2.8.0', 'scikit-learn>=0.24.2', 'pyyaml>=5.4.1', 'numpy>=1.21.2', diff --git a/TabPFN/requirements.txt b/TabPFN/requirements.txt index 10d12ca..3346589 100644 --- a/TabPFN/requirements.txt +++ b/TabPFN/requirements.txt @@ -1,6 +1,6 @@ # Please use python V 3.7 to be compatible with all packages gpytorch==1.5.0 -torch==1.9.0 +torch==2.8.0 scikit-learn==0.24.2 pyyaml==5.4.1 seaborn==0.11.2