wangjin2000 commited on
Commit
04fa7ee
·
verified ·
1 Parent(s): 3936a88

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -122,6 +122,8 @@ def finetune(base_model_path, peptide_length): #, train_dataset, test_dataset)
122
  def compute_pseudo_perplexity(model, tokenizer, protein_seq, binder_seq):
123
  sequence = protein_seq + binder_seq
124
  original_input = tokenizer.encode(sequence, return_tensors='pt').to(model.device)
 
 
125
  length_of_binder = len(binder_seq)
126
 
127
  # Prepare a batch with each row having one masked token from the binder sequence
 
122
  def compute_pseudo_perplexity(model, tokenizer, protein_seq, binder_seq):
123
  sequence = protein_seq + binder_seq
124
  original_input = tokenizer.encode(sequence, return_tensors='pt').to(model.device)
125
+ print("125: original_input:", original_input)
126
+ print("126: original_input.size:", original_input.shape)
127
  length_of_binder = len(binder_seq)
128
 
129
  # Prepare a batch with each row having one masked token from the binder sequence