AlienChen commited on
Commit
30fd543
·
verified ·
1 Parent(s): deb019c

Update models/peptide_classifiers.py

Browse files
Files changed (1) hide show
  1. models/peptide_classifiers.py +1 -1
models/peptide_classifiers.py CHANGED
@@ -680,7 +680,7 @@ class HalfLifeModel:
680
 
681
 
682
  def load_bindevaluator(checkpoint_path, device):
683
- bindevaluator = BindEvaluator.load_from_checkpoint(checkpoint_path, n_layers=8, d_model=128, d_hidden=128, n_head=8, d_k=64, d_v=128, d_inner=64).to(device)
684
  bindevaluator.eval()
685
  for param in bindevaluator.parameters():
686
  param.requires_grad = False
 
680
 
681
 
682
  def load_bindevaluator(checkpoint_path, device):
683
+ bindevaluator = BindEvaluator.load_from_checkpoint(checkpoint_path, weights_only=False, n_layers=8, d_model=128, d_hidden=128, n_head=8, d_k=64, d_v=128, d_inner=64).to(device)
684
  bindevaluator.eval()
685
  for param in bindevaluator.parameters():
686
  param.requires_grad = False