11gb download for prostT5 is too much for ci

lucidrains · lucidrains · commit 3543a323814b · 2024-09-14T10:27:41.000-07:00
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

Original file line number	Diff line number	Diff line change
`@@ -1095,7 +1095,7 @@ def test_alphafold3_with_plm_embeddings():`
`1095`	`1095`	`num_molecule_mods=0,`
`1096`	`1096`	`dim_atom_inputs=77,`
`1097`	`1097`	`dim_template_feats=108,`
`1098`		`- plm_embeddings=("esm2_t33_650M_UR50D", "prostT5"),`
	`1098`	`+ plm_embeddings="esm2_t33_650M_UR50D",`
`1099`	`1099`	`)`
`1100`	`1100`
`1101`	`1101`	`# mock inputs`