detectivejoewest commited on
Commit
08ea9dc
·
verified ·
1 Parent(s): 0285bef

Upload handler.py

Browse files
Files changed (1) hide show
  1. handler.py +4 -1
handler.py CHANGED
@@ -1,4 +1,7 @@
1
- import tiktoken
 
 
 
2
  from typing import Dict, Any
3
  import numpy as np
4
  import tensorflow as tf
 
1
+ from transformers import PreTrainedTokenizerFast
2
+
3
+ token2vec = PreTrainedTokenizerFast.from_pretrained("cl100k_tokenizer")
4
+
5
  from typing import Dict, Any
6
  import numpy as np
7
  import tensorflow as tf