diff --git a/bionemo-recipes/models/geneformer/pyproject.toml b/bionemo-recipes/models/geneformer/pyproject.toml index 2de37ed6b3..168cb7485e 100644 --- a/bionemo-recipes/models/geneformer/pyproject.toml +++ b/bionemo-recipes/models/geneformer/pyproject.toml @@ -16,6 +16,7 @@ dependencies = [ "nemo_toolkit[lightning]==2.4.0", "fiddle", "megatron-core", + "nvidia-resiliency-ext>=0.6.0", "pytest", "requests", "hydra-core",