Spaces:
Runtime error
Runtime error
Commit
·
5997585
1
Parent(s):
8636736
Compile Worked, fix path data
Browse files
oneformer/data/tokenizer.py
CHANGED
@@ -24,6 +24,8 @@
|
|
24 |
# Modified by Jiarui Xu
|
25 |
# -------------------------------------------------------------------------
|
26 |
|
|
|
|
|
27 |
import gzip
|
28 |
import html
|
29 |
import os
|
@@ -36,7 +38,10 @@ import torch
|
|
36 |
|
37 |
@lru_cache()
|
38 |
def default_bpe():
|
39 |
-
|
|
|
|
|
|
|
40 |
|
41 |
|
42 |
@lru_cache()
|
@@ -122,6 +127,9 @@ class SimpleTokenizer(object):
|
|
122 |
def __init__(self, bpe_path: str = default_bpe()):
|
123 |
self.byte_encoder = bytes_to_unicode()
|
124 |
self.byte_decoder = {v: k for k, v in self.byte_encoder.items()}
|
|
|
|
|
|
|
125 |
merges = gzip.open(bpe_path).read().decode('utf-8').split('\n')
|
126 |
merges = merges[1:49152 - 256 - 2 + 1]
|
127 |
merges = [tuple(merge.split()) for merge in merges]
|
|
|
24 |
# Modified by Jiarui Xu
|
25 |
# -------------------------------------------------------------------------
|
26 |
|
27 |
+
from StringIO import StringIO
|
28 |
+
from urllib import urlopen
|
29 |
import gzip
|
30 |
import html
|
31 |
import os
|
|
|
38 |
|
39 |
@lru_cache()
|
40 |
def default_bpe():
|
41 |
+
url = 'https://github.com/SHI-Labs/OneFormer/blob/main/oneformer/data/bpe_simple_vocab_16e6.txt.gz'
|
42 |
+
inmemory = StringIO(urlopen(url).read())
|
43 |
+
return inmemory
|
44 |
+
# return os.path.join(os.path.dirname(os.path.abspath(__file__)), 'bpe_simple_vocab_16e6.txt.gz')
|
45 |
|
46 |
|
47 |
@lru_cache()
|
|
|
127 |
def __init__(self, bpe_path: str = default_bpe()):
|
128 |
self.byte_encoder = bytes_to_unicode()
|
129 |
self.byte_decoder = {v: k for k, v in self.byte_encoder.items()}
|
130 |
+
|
131 |
+
merges = gzip.GzipFile(fileobj=bpe_path, mode='rb')
|
132 |
+
|
133 |
merges = gzip.open(bpe_path).read().decode('utf-8').split('\n')
|
134 |
merges = merges[1:49152 - 256 - 2 + 1]
|
135 |
merges = [tuple(merge.split()) for merge in merges]
|