Skip to content

GitLab

Unverified Commit c3436dd1 authored by oahzxl's avatar oahzxl Committed by GitHub
Browse files

fix indentation (#127)

parent 7d10334b
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
fastfold/model/hub/alphafold.py
View file @ c3436dd1
...@@ -318,7 +318,7 @@ class AlphaFold(nn.Module): ...@@ -318,7 +318,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:
...@@ -352,7 +352,7 @@ class AlphaFold(nn.Module): ...@@ -352,7 +352,7 @@ class AlphaFold(nn.Module):
pair_mask=pair_mask.to(dtype=z[0].dtype), pair_mask=pair_mask.to(dtype=z[0].dtype),
_mask_trans=self.config._mask_trans, _mask_trans=self.config._mask_trans,
)[0] )[0]
del extra_msa_feat, extra_msa_fn del extra_msa_feat, extra_msa_fn
# Run MSA + pair embeddings through the trunk of the network # Run MSA + pair embeddings through the trunk of the network
# m: [*, S, N, C_m] # m: [*, S, N, C_m]
......
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