Update alphafold3.py (#254)

lucidrains · Sep 14, 2024 · c5a694e · c5a694e
1 parent 3543a32
commit c5a694e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/alphafold3_pytorch/alphafold3.py b/alphafold3_pytorch/alphafold3.py
@@ -5959,7 +5959,7 @@ def __init__(
         checkpoint_diffusion_module = False,
         detach_when_recycling = True,
         pdb_training_set=True,
-        plm_embeddings: PLMEmbeddings | tuple[PLMEmbedding, ...] | None = None,
+        plm_embeddings: PLMEmbedding | tuple[PLMEmbedding, ...] | None = None,
         plm_kwargs: dict | tuple[dict, ...] | None = None,
         constraint_embeddings: int | None = None,
     ):