Skip to content

Commit bb4421d

Browse files
authored
remove git install for megatron-fsdp now that new version is released (#1402)
Fixes BIO-9 Signed-off-by: Peter St. John <pstjohn@nvidia.com>
1 parent db7009a commit bb4421d

File tree

5 files changed

+5
-10
lines changed

5 files changed

+5
-10
lines changed

.devcontainer/recipes/requirements.txt

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,7 @@ datasets
33
deepspeed
44
hydra-core
55
lm-eval
6-
# TOT megatron-mfsdp until NVIDIA/Megatron-LM#2575 is in a release.
7-
megatron-fsdp @ git+https://github.com/NVIDIA/Megatron-LM.git@main#subdirectory=megatron/core/distributed/fsdp/src
6+
megatron-fsdp
87
peft
98
pytest
109
torch

bionemo-recipes/models/esm2/pyproject.toml

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,7 @@ dependencies = [
1212
"datasets",
1313
"hydra-core",
1414
"jinja2",
15-
# TOT megatron-mfsdp until NVIDIA/Megatron-LM#2575 is in a release.
16-
"megatron-fsdp @ git+https://github.com/NVIDIA/Megatron-LM.git@main#subdirectory=megatron/core/distributed/fsdp/src",
15+
"megatron-fsdp",
1716
"omegaconf",
1817
"peft",
1918
"pytest",

bionemo-recipes/recipes/esm2_native_te/requirements.txt

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
datasets
22
hydra-core
3-
# TOT megatron-mfsdp until NVIDIA/Megatron-LM#2575 is in a release.
4-
megatron-fsdp @ git+https://github.com/NVIDIA/Megatron-LM.git@main#subdirectory=megatron/core/distributed/fsdp/src
3+
megatron-fsdp
54
pytest
65
torch
76
torchao!=0.14.0

bionemo-recipes/recipes/geneformer_native_te_mfsdp_fp8/requirements.txt

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
datasets
22
hydra-core
3-
# TOT megatron-mfsdp until NVIDIA/Megatron-LM#2575 is in a release.
4-
megatron-fsdp @ git+https://github.com/NVIDIA/Megatron-LM.git@main#subdirectory=megatron/core/distributed/fsdp/src
3+
megatron-fsdp
54
safetensors
65
torch
76
torchao!=0.14.0

bionemo-recipes/recipes/vit/requirements.txt

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,7 @@
11
torch
22
torchvision
33
transformer_engine
4-
# TOT megatron-mfsdp until NVIDIA/Megatron-LM#2575 is in a release.
5-
megatron-fsdp @ git+https://github.com/NVIDIA/Megatron-LM.git@main#subdirectory=megatron/core/distributed/fsdp/src
4+
megatron-fsdp
65
hydra-core
76
einops
87
wandb

0 commit comments

Comments
 (0)