pub_soft/ThermoMPNN
3
0
Fork 0
mirror of https://github.com/Kuhlman-Lab/ThermoMPNN.git synced 2026-06-04 15:04:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

Patched typo in utils tied_sample

Browse Source
This commit is contained in:
Henry Dieckhaus
2024-03-06 10:51:55 -05:00
parent 6a96466a20
commit 5df9778f27
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
protein_mpnn_utils.py
View File

@@ -1393,7 +1393,7 @@ class ProteinMPNN(nn.Module):
h_E = self.W_e(E)
# Encoder is unmasked self-attention
mask_attend = gather_nodes(mask.unsqueeze(-1), E_idx).squeeze(-1)
mask_attend = mask.unsqueeze(-1) * mask_attendE
mask_attend = mask.unsqueeze(-1) * mask_attend
for layer in self.encoder_layers:
h_V, h_E = layer(h_V, h_E, E_idx, mask, mask_attend)