From 3543a323814bd0d30718afb5c51487bf8fd43038 Mon Sep 17 00:00:00 2001 From: lucidrains Date: Sat, 14 Sep 2024 10:27:41 -0700 Subject: [PATCH] 11gb download for prostT5 is too much for ci --- tests/test_af3.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_af3.py b/tests/test_af3.py index ece260a2..08b0672c 100644 --- a/tests/test_af3.py +++ b/tests/test_af3.py @@ -1095,7 +1095,7 @@ def test_alphafold3_with_plm_embeddings(): num_molecule_mods=0, dim_atom_inputs=77, dim_template_feats=108, - plm_embeddings=("esm2_t33_650M_UR50D", "prostT5"), + plm_embeddings="esm2_t33_650M_UR50D", ) # mock inputs