Skip to content

GitLab

Commit b3f8a5db authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Fix undefined reference

parent d59b06b4
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
openfold/model/structure_module.py
View file @ b3f8a5db
...@@ -637,12 +637,12 @@ class StructureModule(nn.Module): ...@@ -637,12 +637,12 @@ class StructureModule(nn.Module):
Returns: Returns:
A dictionary of outputs A dictionary of outputs
""" """
s = evoformer_output_dict["single"]
if mask is None: if mask is None:
# [*, N] # [*, N]
mask = s.new_ones(s.shape[:-1]) mask = s.new_ones(s.shape[:-1])
s = evoformer_output_dict["single"]
# [*, N, C_s] # [*, N, C_s]
s = self.layer_norm_s(s) s = self.layer_norm_s(s)
......
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