11gb download for prostT5 is too much for ci

lucidrains · Sep 14, 2024 · 3543a32 · 3543a32
1 parent 02e3ca2
commit 3543a32
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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