Update ProteinMPNN/vanilla_proteinmpnn/protein_mpnn_utils.py

ProteinMPNN/vanilla_proteinmpnn/protein_mpnn_utils.py

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