Skip to content

Commit fa69bd9

Browse files
committed
add jira ticket
Signed-off-by: Peter St. John <pstjohn@nvidia.com>
1 parent 215052e commit fa69bd9

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

recipes/esm2_accelerate/test_train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -243,7 +243,7 @@ def test_accelerate_launch(accelerate_config, model_tag, tmp_path):
243243
# modeling_esm_te import seems to fix it.
244244
# ("fsdp1_te.yaml", "nvidia/esm2_t6_8M_UR50D"),
245245
("fsdp2_te.yaml", "nvidia/esm2_t6_8M_UR50D"),
246-
# TODO: (BIONEMO-????). These tests were broken by https://github.com/huggingface/transformers/pull/40370, but
246+
# TODO: (BIONEMO-2761). These tests were broken by https://github.com/huggingface/transformers/pull/40370, but
247247
# oddly the single-GPU tests still seem to pass. Changing the attention_backend doesn't seem to help.
248248
# ("default.yaml", "facebook/esm2_t6_8M_UR50D"),
249249
# ("fsdp1_hf.yaml", "facebook/esm2_t6_8M_UR50D"),

0 commit comments

Comments
 (0)