Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/openfold.git" did not exist on "02fc43762b9c9fc91fcf872c5b9f64bdafd9ea2b"
Commit 9b114f28 authored by sachinkadyan7's avatar sachinkadyan7 Committed by Sachin Kadyan
Browse files

Renamed `preembedding_embedder` to `input_embedder`

parent 5a8d2b78
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
openfold/model/model.py
View file @ 9b114f28
...@@ -78,7 +78,7 @@ class AlphaFold(nn.Module): ...@@ -78,7 +78,7 @@ class AlphaFold(nn.Module):
# If using seqemb mode, embed the sequence embeddings passed # If using seqemb mode, embed the sequence embeddings passed
# to the model ("preembeddings") instead of embedding the sequence # to the model ("preembeddings") instead of embedding the sequence
if self.seqemb_mode: if self.seqemb_mode:
self.preembedding_embedder = PreembeddingEmbedder( self.input_embedder = PreembeddingEmbedder(
**self.config["preembedding_embedder"], **self.config["preembedding_embedder"],
) )
else: else:
...@@ -251,7 +251,7 @@ class AlphaFold(nn.Module): ...@@ -251,7 +251,7 @@ class AlphaFold(nn.Module):
# m: [*, 1, N, C_m] # m: [*, 1, N, C_m]
# z: [*, N, N, C_z] # z: [*, N, N, C_z]
if self.seqemb_mode: if self.seqemb_mode:
m, z = self.preembedding_embedder( m, z = self.input_embedder(
feats["target_feat"], feats["target_feat"],
feats["residue_index"], feats["residue_index"],
feats["seq_embedding"] feats["seq_embedding"]
......
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