diff --git a/hf_space/requirements.txt b/hf_space/requirements.txt index c76a8f2..a6efc18 100644 --- a/hf_space/requirements.txt +++ b/hf_space/requirements.txt @@ -1,4 +1,4 @@ gradio>=6.14.0 -torch>=2.1 +torch>=2.12.0 numpy dose diff --git a/pyproject.toml b/pyproject.toml index eea9ba3..ce64f8d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -21,7 +21,7 @@ classifiers = [ "Topic :: Scientific/Engineering :: Artificial Intelligence", ] dependencies = [ - "torch>=2.1", + "torch>=2.12.0", # transformer_lens upper pin is conservative: TL has had breaking # changes in past majors. Loosen when TL 3.x ships and is validated. "transformer_lens>=3.2.1,<4.0",