McClain commited on
Commit
921067b
·
verified ·
1 Parent(s): ee76406

Upload tokenization_plasmid_lm.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. tokenization_plasmid_lm.py +3 -16
tokenization_plasmid_lm.py CHANGED
@@ -56,28 +56,15 @@ class PlasmidLMTokenizer(PreTrainedTokenizer):
56
  if tok in self._vocab:
57
  special_kwargs[name] = tok
58
 
 
 
 
59
  super().__init__(**special_kwargs, **kwargs)
60
 
61
  @property
62
  def vocab_size(self) -> int:
63
  return len(self._vocab)
64
 
65
- @property
66
- def pad_token_id(self) -> int:
67
- return self._vocab.get("<PAD>", 0)
68
-
69
- @property
70
- def bos_token_id(self) -> int:
71
- return self._vocab.get("<BOS>", 1)
72
-
73
- @property
74
- def eos_token_id(self) -> int:
75
- return self._vocab.get("<EOS>", 2)
76
-
77
- @property
78
- def sep_token_id(self) -> int:
79
- return self._vocab.get("<SEP>", 3)
80
-
81
  def get_vocab(self) -> dict:
82
  return dict(self._vocab)
83
 
 
56
  if tok in self._vocab:
57
  special_kwargs[name] = tok
58
 
59
+ # Disable automatic BOS/EOS insertion (transformers 5.x); prompts
60
+ # already contain <BOS>...<SEP> and the model generates <EOS> itself.
61
+ kwargs.setdefault("special_tokens_pattern", "none")
62
  super().__init__(**special_kwargs, **kwargs)
63
 
64
  @property
65
  def vocab_size(self) -> int:
66
  return len(self._vocab)
67
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
68
  def get_vocab(self) -> dict:
69
  return dict(self._vocab)
70