Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
whoops
Browse files- 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 |
-
|
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)
|