Update ProteinMPNN/vanilla_proteinmpnn/protein_mpnn_utils.py

ProteinMPNN/vanilla_proteinmpnn/protein_mpnn_utils.py

@@ -733,7 +733,7 @@ class ProteinFeatures(nn.Module):
         return RBF_A_B
 
     def forward(self, X, mask, residue_idx, chain_labels):
-        if self.augment_eps > 0:
+        if int(self.augment_eps) > 0:
             X = X + self.augment_eps * torch.randn_like(X)
         
         b = X[:,:,1,:] - X[:,:,0,:]