Compile Worked, fix path data

Dockerfile

@@ -31,32 +31,32 @@ RUN chmod -R 777 $WORKDIR
 
 COPY requirements.txt $WORKDIR/requirements.txt
 COPY . .
-# RUN cd $WORKDIR/oneformer/data/bpe_simple_vocab_16e6.txt.gz && wget https://github.com/SHI-Labs/OneFormer/blob/main/oneformer/data/bpe_simple_vocab_16e6.txt.gz && ls && cd $WORKDIR
+RUN cd $WORKDIR/oneformer/data && ls
 
-RUN pip install --no-cache-dir --upgrade -r $WORKDIR/requirements.txt
+# RUN pip install --no-cache-dir --upgrade -r $WORKDIR/requirements.txt
 
-ARG TORCH_CUDA_ARCH_LIST=7.5+PTX
+# ARG TORCH_CUDA_ARCH_LIST=7.5+PTX
 
 
-RUN pip install ninja
+# RUN pip install ninja
 
-USER root
-RUN chown -R user:user /usr
-RUN chmod -R 777 /usr
-RUN chown -R user:user $HOME
-RUN chmod -R 777 $HOME
-RUN chown -R user:user $WORKDIR
-RUN chmod -R 777 $WORKDIR
+# USER root
+# RUN chown -R user:user /usr
+# RUN chmod -R 777 /usr
+# RUN chown -R user:user $HOME
+# RUN chmod -R 777 $HOME
+# RUN chown -R user:user $WORKDIR
+# RUN chmod -R 777 $WORKDIR
 
-USER user
-RUN ln -s $WORKDIR/oneformer/modeling/pixel_decoder/ops/ $WORKDIR/ && ls && cd ops/ && FORCE_CUDA=1 python setup.py build --build-base=$WORKDIR/ install --user && cd ..
-RUN sh deform_setup.sh
+# USER user
+# RUN ln -s $WORKDIR/oneformer/modeling/pixel_decoder/ops/ $WORKDIR/ && ls && cd ops/ && FORCE_CUDA=1 python setup.py build --build-base=$WORKDIR/ install --user && cd ..
+# RUN sh deform_setup.sh
 
-USER user
-RUN sh deform_setup.sh
+# USER user
+# RUN sh deform_setup.sh
 
-USER user
+# USER user
 
-EXPOSE 7860
+# EXPOSE 7860
 
-ENTRYPOINT ["python", "gradio_app.py"]
+# ENTRYPOINT ["python", "gradio_app.py"]

oneformer/data/tokenizer.py

@@ -24,8 +24,6 @@
 # Modified by Jiarui Xu
 # -------------------------------------------------------------------------
 
-# from StringIO import StringIO
-# from urllib import urlopen
 import wget
 import gzip
 import html
@@ -41,11 +39,8 @@ import torch
 def default_bpe():
     url = 'https://github.com/SHI-Labs/OneFormer/blob/main/oneformer/data/bpe_simple_vocab_16e6.txt.gz'
     wget.download(url, out=os.path.join(os.path.dirname(os.path.abspath(__file__)), 'bpe_simple_vocab_16e6.txt.gz'))
-    # inmemory = StringIO(urlopen(url).read())
-    # return inmemory
     return os.path.join(os.path.dirname(os.path.abspath(__file__)), 'bpe_simple_vocab_16e6.txt.gz')
 
-
 @lru_cache()
 def bytes_to_unicode():
     """Returns list of utf-8 byte and a corresponding list of unicode strings.
@@ -129,10 +124,6 @@ class SimpleTokenizer(object):
     def __init__(self, bpe_path: str = default_bpe()):
         self.byte_encoder = bytes_to_unicode()
         self.byte_decoder = {v: k for k, v in self.byte_encoder.items()}
-
-        print("-----------\n")
-        print(bpe_path)
-        print("-----------\n")
         merges = gzip.open(bpe_path).read().decode('utf-8').split('\n')
         merges = merges[1:49152 - 256 - 2 + 1]
         merges = [tuple(merge.split()) for merge in merges]