praeclarumjj3 commited on
Commit
5997585
·
1 Parent(s): 8636736

Compile Worked, fix path data

Browse files
Files changed (1) hide show
  1. oneformer/data/tokenizer.py +9 -1
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
- return os.path.join(os.path.dirname(os.path.abspath(__file__)), 'bpe_simple_vocab_16e6.txt.gz')
 
 
 
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]