diff --git a/bionemo-recipes/models/amplify/pyproject.toml b/bionemo-recipes/models/amplify/pyproject.toml index 5bc261da76..2aec67e4aa 100644 --- a/bionemo-recipes/models/amplify/pyproject.toml +++ b/bionemo-recipes/models/amplify/pyproject.toml @@ -17,7 +17,7 @@ dependencies = [ "omegaconf", "pytest", "setuptools<82", - "torch==2.6.0a0+ecf3bae40a.nv25.01", + "torch==2.8.0", "transformer_engine[pytorch]", "transformers<5.0", # TODO(BIO-143): update AMPLIFY to support Transformers v5 "xformers",