Benjamin Hoover commited on
Commit
4a48eda
1 Parent(s): fad2f31
Files changed (1) hide show
  1. server/model_api.py +1 -2
server/model_api.py CHANGED
@@ -54,8 +54,7 @@ class ModelDetails:
54
  clst = self.tok.cls_token_id
55
  sept = self.tok.sep_token_id
56
  if (bost is not None) and (bost != clst)and add_special_tokens:
57
- if (clst is not None and clst != sept):
58
- ids.insert(0, bost)
59
 
60
  inputs = self.tok.prepare_for_model(ids, add_special_tokens=add_special_tokens, return_tensors="pt")
61
  parsed_input = self.parse_inputs(inputs, mask_attentions=mask_attentions)
54
  clst = self.tok.cls_token_id
55
  sept = self.tok.sep_token_id
56
  if (bost is not None) and (bost != clst)and add_special_tokens:
57
+ ids.insert(0, bost)
 
58
 
59
  inputs = self.tok.prepare_for_model(ids, add_special_tokens=add_special_tokens, return_tensors="pt")
60
  parsed_input = self.parse_inputs(inputs, mask_attentions=mask_attentions)