Skip to content

GitLab

Commit b6f5ac33 authored by zhuww's avatar zhuww
Browse files

fix the problem of reasoning without templates

parent f5113baf
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
fastfold/model/hub/alphafold.py
View file @ b6f5ac33
...@@ -324,7 +324,7 @@ class AlphaFold(nn.Module): ...@@ -324,7 +324,7 @@ class AlphaFold(nn.Module):
[feats["msa_mask"], template_embeds["template_mask"]], [feats["msa_mask"], template_embeds["template_mask"]],
dim=-2, dim=-2,
) )
del template_feats, template_embeds del template_feats, template_embeds
# Embed extra MSA features + merge with pairwise embeddings # Embed extra MSA features + merge with pairwise embeddings
if self.config.extra_msa.enabled: if self.config.extra_msa.enabled:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment