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

fix indentation (#127)

parent 7d10334b
......@@ -318,7 +318,7 @@ class AlphaFold(nn.Module):
[feats["msa_mask"], template_embeds["template_mask"]],
dim=-2,
)
del template_feats, template_embeds
del template_feats, template_embeds
# Embed extra MSA features + merge with pairwise embeddings
if self.config.extra_msa.enabled:
......@@ -352,7 +352,7 @@ class AlphaFold(nn.Module):
pair_mask=pair_mask.to(dtype=z[0].dtype),
_mask_trans=self.config._mask_trans,
)[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
# 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